Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, LiteLLMModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -13,7 +13,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+        model = LiteLLMModel(model_id = "gemini/gemini-2.0-flash",
+                             api_key = os.getenv("GEMINI_API_KEY"),
+                             max_tokens=8192,)
         self.agent = CodeAgent(
             model = model,
             tools=[DuckDuckGoSearchTool()]