bstraehle commited on
Commit
1ded0b3
·
verified ·
1 Parent(s): 0c53631

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -21,9 +21,9 @@ def invoke(level, question, file_name, ground_truth, openai_api_key):
21
 
22
  with lock:
23
  os.environ["OPENAI_API_KEY"] = openai_api_key
24
- answer = run_agent(level, question, file_name)
25
  del os.environ["OPENAI_API_KEY"]
26
- return answer, str(answer == ground_truth)
27
 
28
  gr.close_all()
29
 
@@ -34,6 +34,7 @@ demo = gr.Interface(fn = invoke,
34
  gr.Textbox(label = "Ground Truth"),
35
  gr.Textbox(label = "OpenAI API Key", type = "password")],
36
  outputs = [gr.Textbox(label = "Answer", lines = 1, interactive = False),
 
37
  gr.Radio(["True", "False"], label = "Matches Ground Truth", interactive = False)],
38
  title = "General AI Assistant (GAIA)",
39
  description = os.environ["DESCRIPTION"],
 
21
 
22
  with lock:
23
  os.environ["OPENAI_API_KEY"] = openai_api_key
24
+ answer, final_answer = run_agent(level, question, file_name)
25
  del os.environ["OPENAI_API_KEY"]
26
+ return answer, final_answer, str(final_answer == ground_truth)
27
 
28
  gr.close_all()
29
 
 
34
  gr.Textbox(label = "Ground Truth"),
35
  gr.Textbox(label = "OpenAI API Key", type = "password")],
36
  outputs = [gr.Textbox(label = "Answer", lines = 1, interactive = False),
37
+ gr.Textbox(label = "Final Answer", lines = 1, interactive = False),
38
  gr.Radio(["True", "False"], label = "Matches Ground Truth", interactive = False)],
39
  title = "General AI Assistant (GAIA)",
40
  description = os.environ["DESCRIPTION"],