Update app.py

app.py

@@ -79,7 +79,13 @@ final_answer = FinalAnswerTool()
 # custom_role_conversions=None,
 # )
 
-model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="gemini_api")) 
+#model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="gemini_api"))
+
+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,
+    )
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)