Update app.py

app.py

@@ -12,8 +12,11 @@ import yaml
 
 # Import modules and classes from openai and smolagents
 # import openai
-from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool
 from tools import SimplifyAnswerTool
+from huggingface_hub import login, InferenceClient
+
+login("HF_API_KEY")
 
 # Set API keys
 # os.environ["OPENAI_API_KEY"] = "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"
@@ -22,7 +25,7 @@ from tools import SimplifyAnswerTool
 
 # Create model
 # model = OpenAIServerModel(model_id="gpt-4o")
-model = HfApiModel(
+model = InferenceClient(
 max_tokens=2096,
 temperature=0.5,
 # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded