Update app.py

app.py

@@ -1,17 +1,26 @@
 import gradio as gr
 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
 
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0,
-    model_id='mistralai/Mistral-7B-Instruct-v0.3',
-    custom_role_conversions=None,
-)
+#model = HfApiModel(
+#    max_tokens=2096,
+#    temperature=0,
+#    model_id='mistralai/Mistral-7B-Instruct-v0.3',
+#    custom_role_conversions=None,
+#)
+
+
+hf_api = HfApi()
+secrets = hf_api.get_secrets() 
+
+model = LiteLLMModel(
+    model_id="gemini/gemini-2.0-flash-lite", 
+    api_key=secrets.get("GEMINI_API_TOKEN")
+) 
 
 # Initialize the web search tool
 search_tool = DuckDuckGoSearchTool()