Update main.py

main.py

@@ -17,9 +17,16 @@ class ResearchResponse(BaseModel):
     sources: list[str]
     tools_used: list[str]
 
+    
+
 # ✅ Use HuggingFace Mistral model (free)
+
+if hf_token is None:
+    raise ValueError("Missing Hugging Face API token. Please set the 'HUGGINGFACEHUB_API_TOKEN' secret in your HF space settings.")
+
 llm = HuggingFaceHub(
     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+    huggingfacehub_api_token=hf_token,
     model_kwargs={"temperature": 0.7, "max_new_tokens": 500},
 )