Update app.py

app.py

@@ -7,6 +7,7 @@ from tools.final_answer import FinalAnswerTool
 
 from smolagents import GradioUI
 from smolagents import LiteLLMModel
+import os
 
 import litellm
 
@@ -71,13 +72,8 @@ final_answer = FinalAnswerTool()
 # model_id='meta-llama/Llama-3.1-8B-Instruct',# it is possible that this model may be overloaded
 # custom_role_conversions=None,
 # )
-model = LiteLLMModel(
-        model_id="ollama_chat/qwen2:7b",  # Or try other Ollama-supported models
-        api_base="http://127.0.0.1:11434",  # Default Ollama local server
-        num_ctx=8192,
-    )
-
 
+model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="gemini_api")) 
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)