Add gemini model

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, VisitWebpageTool, FinalAnswerTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, OpenAIServerModel, VisitWebpageTool, FinalAnswerTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -13,8 +13,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
+        # self.model = HfApiModel(model="Qwen/Qwen2.5-VL-72B-Instruct"),
+        self.moedl = model = OpenAIServerModel(
+            model_id="gemini-2.5-flash-preview-05-20",
+            api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
+            api_key=os.environ["GEMINI_API_KEY"],
+        )
         self.agent = CodeAgent(
-            model=HfApiModel(model="google/gemma-3-27b-it"),
+            model=self.model,
             tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), FinalAnswerTool()],
             max_steps=5,
         )