Update app.py

app.py

@@ -1,5 +1,14 @@
+try:
+    from smolagents import InferenceClientModel as ModelClass
+except ImportError:
+    try:
+        from smolagents import HfApiModel as ModelClass
+    except ImportError:
+        from smolagents import HfEngine as ModelClass
+
+
+
 from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
-from smolagents import HfApiModel
 import datetime
 import requests
 import pytz
@@ -62,7 +71,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 final_answer = FinalAnswerTool()
 
 # Initialize model
-model = HfApiModel(
+model = ModelClass(
     max_tokens=2096,
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',