bstraehle commited on
Commit
7c614dd
·
verified ·
1 Parent(s): 88d86ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from agent import run_agent
5
 
6
  lock = threading.Lock()
7
 
8
- LLM = "gpt-4o"
9
 
10
  FILE_PATH = "https://huggingface.co/datasets/gaia-benchmark/GAIA/blob/main/2023/validation/"
11
 
@@ -18,7 +18,7 @@ def invoke(level, question, file_name, ground_truth, openai_api_key):
18
 
19
  with lock:
20
  os.environ["OPENAI_API_KEY"] = openai_api_key
21
- result = run_agent(LLM, task)
22
  del os.environ["OPENAI_API_KEY"]
23
  return result
24
 
 
5
 
6
  lock = threading.Lock()
7
 
8
+ MODEL_ID = "gpt-4o"
9
 
10
  FILE_PATH = "https://huggingface.co/datasets/gaia-benchmark/GAIA/blob/main/2023/validation/"
11
 
 
18
 
19
  with lock:
20
  os.environ["OPENAI_API_KEY"] = openai_api_key
21
+ result = run_agent(MODEL_ID, level, question, file_name)
22
  del os.environ["OPENAI_API_KEY"]
23
  return result
24