Update app.py

app.py

@@ -4,14 +4,16 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, PythonInterpreterTool, HfApiModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, PythonInterpreterTool, HfApiModel, LiteLLMModel
 from huggingface_hub import login
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-HF_TOKEN = os.getenv("HF_TOKEN")
-login(HF_TOKEN)
+#HF_TOKEN = os.getenv("HF_TOKEN")
+#login(HF_TOKEN)
+model = LiteLLMModel(model_id="models/gemini-2.0-flash-lite",
+                     api_key=os.getenv("GEMINI_API_KEY"))
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -19,7 +21,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         # Set up model
-        self.model = HfApiModel()
+        self.model = model
 
         # Initialize custom tools
         self.tools = [