Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool, OpenAIServerModel
 import datetime
 import requests
 import pytz
@@ -35,6 +35,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
+# https://docs.litellm.ai/
 #model = LiteLLMModel(
 #  model_id="gemini/gemini-2.0-flash-exp",
 #  max_tokens=2096,
@@ -42,12 +43,12 @@ final_answer = FinalAnswerTool()
 #  api_key=os.getenv("LITELLM_API_KEY")
 #)
 
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
-)
+#model = HfApiModel(
+#    max_tokens=2096,
+#    temperature=0.5,
+#    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+#    custom_role_conversions=None,
+#)
 
 #model = HfApiModel(
 #    max_tokens=2096,
@@ -57,13 +58,14 @@ model = HfApiModel(
 #)
 
 # ollama
-# model = LiteLLMModel(
-#   model_id="ollama_chat/deepseek-r1:7b",
-#   max_tokens=2096,
-#   temperature=0.6,
-#   api_base="http://localhost:11434",
-#   num_ctx=8192
-# )
+model = OpenAIServerModel(
+   model_id="gemma2:2b", #"deepseek-coder-v2:16b",
+   max_tokens=2096,
+   temperature=0.6,
+   api_base="http://localhost:11434",
+   #num_ctx=8192,
+   api_key="",
+)
 
 # transformer
 # model = TransformersModel(
@@ -74,10 +76,9 @@ model = HfApiModel(
 #    temperature=0.6,
 #)
 
-print("Model Image")
-
+# print("Model Image")
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)