Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool, OpenAIServerModel
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool, TransformersModel
 import datetime
 import requests
 import pytz
@@ -58,23 +58,23 @@ final_answer = FinalAnswerTool()
 #)
 
 # ollama
-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="",
-)
+#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(
-#    model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
-#    device_map="auto",
-#    torch_dtype="auto",
-#    max_new_tokens=2096,
-#    temperature=0.6,
-#)
+model = TransformersModel(
+    model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
+    device_map="auto",
+    torch_dtype="auto",
+    max_new_tokens=2096,
+    temperature=0.6,
+)
 
 # print("Model Image")
 # Import tool from Hub