Update agent.py

agent.py

@@ -47,13 +47,12 @@ def build_agent():
     #     huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
     # )
     
-    llm = HuggingFaceHub(
-    repo_id = "question-answering", model="deepset/roberta-base-squad2",
-    model_kwargs={
-        "temperature": 0.7,
-        "max_new_tokens": 100
-    }
-     )
+    llm = HuggingFaceHub(repo_id="distilbert/distilgpt2", task="text-generation",
+                    model_kwargs={
+        "temperature": 0.7,         # Controls randomness (0.0 = deterministic)
+        "max_new_tokens": 100       # Optional: max output length
+    }, 
+    verbose=True)
     
     tools = [use_wikipedia_tool, use_search_tool]