Update app.py

app.py

@@ -11,7 +11,7 @@ import yaml
 # os.system('pip install openai')
 
 # Import modules and classes from openai and smolagents
-from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
+from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
 from tools import SimplifyAnswerTool
 from huggingface_hub import login
 
@@ -20,11 +20,11 @@ HF_API_KEY = os.getenv("HF_API_KEY")
 login(HF_API_KEY)
 
 # Set API key for OPENAI
-os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
+# os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 
 # Create model
-model = OpenAIServerModel(model_id="gpt-4o")
-# model = HfApiModel()
+# model = OpenAIServerModel(model_id="gpt-4o")
+model = HfApiModel()
 
 # Create final answer instance
 final_answer = SimplifyAnswerTool(model)