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, TransformersModel
 import datetime
 import requests
 import pytz
@@ -37,11 +37,29 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
-    custom_role_conversions=None,
+#model = HfApiModel(
+#    max_tokens=2096,
+#    temperature=0.5,
+#    model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
+#    custom_role_conversions=None,
+#)
+
+# ollama
+# model = LiteLLMModel(
+#   model_id="ollama_chat/deepseek-r1:7b",
+#   max_tokens=2096,
+#   temperature=0.6,
+#   api_base="http://localhost:11434",
+#   num_ctx=8192
+# )
+
+# 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,
 )
 
 # Import tool from Hub