Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, TransformersModel
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
@@ -33,9 +33,21 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+final_answer = FinalAnswerTool()
 
+#model = LiteLLMModel(
+#  model_id="gemini/gemini-2.0-flash-exp",
+#  max_tokens=2096,
+#  temperature=0.6,
+#  api_key=os.getenv("LITELLM_API_KEY")
+#)
 
-final_answer = FinalAnswerTool()
+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,
@@ -54,13 +66,13 @@ final_answer = FinalAnswerTool()
 # )
 
 # 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,
+#)
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)