Update app.py

app.py

@@ -11,12 +11,12 @@ import yaml
 # os.system('pip install openai')
 
 # Import modules and classes from openai and smolagents
-import openai
-from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
+# import openai
+from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
 from tools import FinalAnswerTool
 
 # Set API keys
-os.environ["OPENAI_API_KEY"] = "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"
+# os.environ["OPENAI_API_KEY"] = "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"
 # openai.api_key = ('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')
 # GoogleSearch.SERP_API_KEY = 'a2fa0feab384c372147075f190fe5878300bbbb693eb0aced535e5d58189ad70'
 
@@ -24,7 +24,14 @@ os.environ["OPENAI_API_KEY"] = "sk-proj-TEJQqmRT2YncQDpEcTh3mFKG8Pu_lsZH_nEV_dzC
 final_answer = FinalAnswerTool()
 
 # Create model
-model = OpenAIServerModel(model_id="gpt-4o")
+# model = OpenAIServerModel(model_id="gpt-4o")
+model = HfApiModel(
+max_tokens=2096,
+temperature=0.5,
+# model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+custom_role_conversions=None,
+)
 
 # Create prompt template
 with open("prompts.yaml", 'r') as stream: