Update app.py

app.py

@@ -5,7 +5,7 @@ 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
-from huggingface_hub import HfApi
+import os
 
 #model = HfApiModel(
 #    max_tokens=2096,
@@ -15,12 +15,10 @@ from huggingface_hub import HfApi
 #)
 
 
-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")
+    api_key=os.environ['GEMINI_API_TOKEN']
 ) 
 
 # Initialize the web search tool