Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import TransformersModel, DuckDuckGoSearchTool, CodeAgent
+from smolagents import TransformersModel, DuckDuckGoSearchTool, CodeAgent, LiteLLMModel
 
 
 # (Keep Constants as is)
@@ -15,14 +15,20 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("Initializing BasicAgent.")
-        model_id = "Qwen/Qwen2.5-7B-Instruct-1M"
+        model_id = "gemini/gemini-2.0-flash" # "Qwen/Qwen2.5-7B-Instruct-1M"
 
-        self.model = TransformersModel(
+        self.model = LiteLLMModel(
             model_id=model_id,
-            max_new_tokens=4096,
-            device_map="auto"
+            api_key=os.environ.get('GOOGLE_API_KEY')
         )
 
+
+        # self.model = TransformersModel(
+        #     model_id=model_id,
+        #     max_new_tokens=4096,
+        #    device_map="auto"
+        #)
+
         search_tool = DuckDuckGoSearchTool()
 
         self.agent = CodeAgent(