Update app.py

app.py

@@ -7,11 +7,20 @@ from transformers import pipeline
 # Configure logging
 logging.basicConfig(level=logging.DEBUG)
 
-# Set Hugging Face token
-os.environ['HUGGINGFACEHUB_API_TOKEN'] = hf_token
-# Load Hugging Face model
+
+# Retrieve Hugging Face API token from environment variables
+HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+if not HUGGINGFACEHUB_API_TOKEN:
+    logging.error("Hugging Face API token is missing or invalid.")
+    st.error("Hugging Face API token is missing or invalid.")
+    st.stop()
+
+# Now you can use HUGGINGFACEHUB_API_TOKEN in your code, for example:
+from transformers import pipeline
+
+# Load the model using the Hugging Face API token
 MODEL_NAME = "mistralai/Mistral-Large-Instruct-2407"
-llm_pipeline = pipeline("text-generation", model=MODEL_NAME)
+llm_pipeline = pipeline("text-generation", model=MODEL_NAME, use_auth_token=HUGGINGFACEHUB_API_TOKEN)
 
 # Load datasets
 def load_datasets():