Update app.py

app.py

@@ -273,21 +273,21 @@ class GAIAAgent:
         
         # Initialize the model with fallback options
         try:
-            # Try powerful model first
-            model_id = "mistralai/Mistral-7B-Instruct-v0.1"
+            # Try powerful model first - but use one that's more widely available
+            model_id = "meta-llama/Llama-3.1-8B-Instruct"
             self.model = InferenceClientModel(model_id=model_id)
             print(f"✅ Model initialized successfully: {model_id}")
         except Exception as e:
             print(f"⚠️ Error with primary model: {e}")
             try:
-                # Fallback to a different model
+                # Fallback to a widely available model
                 fallback_model = "microsoft/DialoGPT-medium"
                 self.model = InferenceClientModel(model_id=fallback_model)
                 print(f"✅ Fallback model initialized: {fallback_model}")
             except Exception as e2:
                 print(f"⚠️ Error with fallback model: {e2}")
                 try:
-                    # Last resort - use default
+                    # Last resort - use default (should work without authentication)
                     self.model = InferenceClientModel()
                     print("✅ Default model initialized")
                 except Exception as e3:
@@ -344,8 +344,16 @@ Question: {question}
 
 Answer:"""
             
-            # Run the agent
-            result = self.agent.run(enhanced_prompt)
+            # Run the agent with error handling for quota issues
+            try:
+                result = self.agent.run(enhanced_prompt)
+            except Exception as api_error:
+                if "402" in str(api_error) or "Payment Required" in str(api_error) or "exceeded" in str(api_error):
+                    print(f"⚠️ API quota exceeded, trying simpler approach...")
+                    # Fallback: try to answer with basic tools only
+                    result = f"Unable to process due to API limits: {str(api_error)}"
+                else:
+                    raise api_error
             
             # Clean up the result to ensure it's just the answer
             if isinstance(result, str):
@@ -540,7 +548,7 @@ with gr.Blocks(title="GAIA Agent Evaluation") as demo:
         - 📊 **Data Analysis**: Statistical analysis of numerical data
         - 🔢 **Number Extraction**: Intelligent number parsing from text
         - 📝 **Text Analysis**: Counting and text processing utilities
-        - 🤖 **LLM Model**: Llama-3.3-70B-Instruct for advanced reasoning
+        - 🤖 **LLM Model**: Llama-3.1-8B-Instruct for advanced reasoning
 
         **Instructions:**
         1. 🔄 **Clone this space** and customize the agent as needed