Update app.py

app.py

@@ -2,10 +2,9 @@ import os
 import gradio as gr
 from PIL import Image
 from smolagents import (
-    CodeAgent, 
-    InferenceClientModel,       
-    DuckDuckGoSearchTool, 
-   # visit_webpage,
+    CodeAgent,
+    LiteLLMModel,          # <-- NEW: for Gemini
+    DuckDuckGoSearchTool,
     Tool
 )
 import requests
@@ -66,11 +65,17 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             image_generation_tool
         ]
         
-        # 5. Initialize HfApiModel with the top-tier open-source GAIA benchmark model
-        # Qwen2.5-72B-Instruct provides elite multi-step execution logic over Serverless API infrastructure.
-        model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
+        # 5. Initialize Google Gemini 2.0 Flash model using LiteLLMModel
+        #    The API key must be stored as a secret named GEMINI_API_KEY in the HF Space.
+        gemini_api_key = os.getenv("GEMINI_API_KEY")
+        if not gemini_api_key:
+            raise ValueError("GEMINI_API_KEY environment variable not set. Please add it to Hugging Face Space secrets.")
+        model = LiteLLMModel(
+            model_id="gemini/gemini-2.0-flash",
+            api_key=gemini_api_key
+        )
         
-        # 6. Initialize the CodeAgent with the Serverless API backbone
+        # 6. Initialize the CodeAgent with the Gemini model
         agent = CodeAgent(
             tools=all_tools,
             model=model,