Update agent.py

agent.py

@@ -46,9 +46,8 @@ def build_agent():
     #     huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
     # )
     
-    llm = HuggingFaceHub(
-    repo_id="google/flan-t5-base",
-    model_kwargs={"temperature": 0.2, "max_new_tokens": 256})
+    llm = HuggingFaceHub(repo_id="google/flan-t5-base",
+    task="text2text-generation")
     
     tools = [use_wikipedia_tool, use_search_tool]