Update agent.py

agent.py

@@ -53,7 +53,7 @@ def build_agent():
         "temperature": 0.7,
         "max_new_tokens": 100
     },
-    task="text-generation"
+    task="question-answering"
      )
     
     tools = [use_wikipedia_tool, use_search_tool]