thatting commited on
Commit
da23b07
·
verified ·
1 Parent(s): 027fe19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,16 +11,16 @@ import yaml
11
  # os.system('pip install openai')
12
 
13
  # Import modules and classes from openai and smolagents
14
- from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
15
  from tools import SimplifyAnswerTool
16
  from huggingface_hub import login
17
 
18
  # Log in to Hugging Face with API_KEY
19
- HF_API_KEY = os.getenv('HF_API_KEY')
20
  login(HF_API_KEY)
21
 
22
  # Set API key for OPENAI
23
- os.environ['OPENAI_API_KEY"'] = os.getenv('OPENAI_API_KEY')
24
 
25
  # Create model
26
  model = OpenAIServerModel(model_id="gpt-4o")
 
11
  # os.system('pip install openai')
12
 
13
  # Import modules and classes from openai and smolagents
14
+ from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
15
  from tools import SimplifyAnswerTool
16
  from huggingface_hub import login
17
 
18
  # Log in to Hugging Face with API_KEY
19
+ HF_API_KEY = os.getenv("HF_API_KEY")
20
  login(HF_API_KEY)
21
 
22
  # Set API key for OPENAI
23
+ os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
24
 
25
  # Create model
26
  model = OpenAIServerModel(model_id="gpt-4o")