Update agent.py

agent.py

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