Update app.py

app.py

@@ -1,5 +1,16 @@
+import os
 import streamlit as st
 from langchain_community.llms import HuggingFaceHub
+from dotenv import load_dotenv
+
+# Load .env if running locally
+load_dotenv()
+
+# Set your Hugging Face API token (must be set in Hugging Face Spaces Secrets for online use)
+HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+if not HUGGINGFACE_API_TOKEN:
+    st.error("⚠️ Hugging Face API token not found. Please add it as a secret with name 'HUGGINGFACEHUB_API_TOKEN'.")
+    st.stop()
 
 # Models for each task
 SUMMARY_MODEL = "google/flan-t5-small"
@@ -11,7 +22,8 @@ def get_llm(model_id):
     return HuggingFaceHub(
         repo_id=model_id,
         model_kwargs={"temperature": 0.5, "max_new_tokens": 150},
-        task="text2text-generation"
+        task="text2text-generation",
+        huggingfacehub_api_token=HUGGINGFACE_API_TOKEN
     )
 
 # Streamlit app UI