Update app.py

app.py

@@ -1,5 +1,5 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-from smolagents.models import TransformersModel
+# from smolagents.models import TransformersModel
 import datetime
 import requests
 import pytz
@@ -92,12 +92,13 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 
 final_answer = FinalAnswerTool()
-# model = HfApiModel(
-# max_tokens=2096,
-# temperature=0.5,
-# model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
-# custom_role_conversions=None,
-# )
+model = HfApiModel(
+max_tokens=2096,
+temperature=0.5,
+#model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',
+model_id="https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud/",
+custom_role_conversions=None,
+)
 
 engine = TransformersModel(
     model_id="meta-llama/Llama-3.2-3B-Instruct",
@@ -115,7 +116,7 @@ with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
-    model=engine,
+    model=model,
     tools=[
         final_answer, 
         get_current_time_in_timezone,