Update agent.py

agent.py

@@ -1,13 +1,14 @@
-from smolagents import CodeAgent, HfApiModel
+from smolagents import CodeAgent, InferenceClientModel
 from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
 
 
 class GaiaAgent:
     def __init__(self):
-        self.model = HfApiModel(
-            max_tokens = 2096,
+        self.model = InferenceClientModel(
+            max_tokens = 8096,
             temperature = 0.5, 
-            model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+            model_id = "deepseek-ai/DeepSeek-R1-0528"
+            provider = "huggingface"         #'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
             custom_role_conversations=None
         )
 #or