Update agent.py

agent.py

@@ -47,12 +47,16 @@ def build_agent():
     #     huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
     # )
     
-    llm = ChatHuggingFace(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 ))
+    llm = ChatHuggingFace(
+    HuggingFaceHub(
+        repo_id="microsoft/DialoGPT-medium",
+        task="text-generation",
+        model_kwargs={
+            "temperature": 0.7,
+            "max_new_tokens": 100
+        }
+    ),
+    verbose=True)
     
     tools = [use_wikipedia_tool, use_search_tool]