Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 import time
-from smolagents import TransformersModel, DuckDuckGoSearchTool, CodeAgent, LiteLLMModel
+from smolagents import DuckDuckGoSearchTool, CodeAgent, LiteLLMModel
 
 
 # (Keep Constants as is)
@@ -16,20 +16,13 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("Initializing BasicAgent.")
-        model_id = "gemini/gemini-2.0-flash-lite" # "Qwen/Qwen2.5-7B-Instruct-1M"
+        model_id = "gemini/gemini-2.0-flash-live-001" 
 
         self.model = LiteLLMModel(
             model_id=model_id,
             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(