Replace the default Hugging Face model with Gemini-2.0-flash-lite

app.py

@@ -1,13 +1,17 @@
 import gradio as gr
+import os
 import random
-from smolagents import GradioUI, CodeAgent, HfApiModel
+from smolagents import GradioUI, CodeAgent, HfApiModel, LiteLLMModel
 
 # Import our custom tools from their modules
 from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
 from retriever import load_guest_dataset
 
 # Initialize the Hugging Face model
-model = HfApiModel()
+model = LiteLLMModel(
+    model_id="gemini/gemini-2.0-flash-lite",
+    api_key=os.getenv("GEMINI_API_TOKEN")
+)
 
 # Initialize the web search tool
 search_tool = DuckDuckGoSearchTool()