Last approach

app.py

@@ -237,11 +237,19 @@ class GAIAAgent:
     def __init__(self):
         print("Initializing GAIA Agent...")
         
-        # Initialize model with InferenceClientModel instead of HfApiModel
-        self.model = InferenceClientModel(
-            model_id="microsoft/DialoGPT-medium",
-            token=os.getenv("HUGGINGFACE_INFERENCE_TOKEN")
-        )
+        # Initialize model with InferenceClientModel
+        try:
+            # Use a more capable model for the agent
+            self.model = InferenceClientModel(
+                model_id="microsoft/DialoGPT-medium",
+                token=os.getenv("HUGGINGFACE_INFERENCE_TOKEN")
+            )
+        except Exception as e:
+            print(f"Error initializing model: {e}")
+            # Fallback to a simpler approach if the model fails
+            self.model = InferenceClientModel(
+                model_id="microsoft/DialoGPT-medium"
+            )
         
         # Custom tools list
         custom_tools = [