Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import os, threading
|
|
4 |
from crew import run_crew
|
5 |
from util import get_questions
|
6 |
|
7 |
-
|
8 |
|
9 |
def invoke(level, question, file_name, ground_truth, openai_api_key):
|
10 |
if not question:
|
@@ -13,11 +13,14 @@ def invoke(level, question, file_name, ground_truth, openai_api_key):
|
|
13 |
if not openai_api_key:
|
14 |
raise gr.Error("OpenAI API Key is required.")
|
15 |
|
|
|
|
|
16 |
with lock:
|
17 |
answer = ""
|
18 |
|
19 |
try:
|
20 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
|
|
21 |
answer = run_crew(question, f"data/{file_name}")
|
22 |
except Exception as e:
|
23 |
raise gr.Error(e)
|
@@ -37,7 +40,7 @@ demo = gr.Interface(fn=invoke,
|
|
37 |
outputs=[gr.Textbox(label="Answer", lines=1, interactive=False)],
|
38 |
title="General AI Assistant (GAIA) π€π€π€",
|
39 |
description=os.environ["DESCRIPTION"],
|
40 |
-
examples=get_questions(),
|
41 |
cache_examples=False
|
42 |
)
|
43 |
|
|
|
4 |
from crew import run_crew
|
5 |
from util import get_questions
|
6 |
|
7 |
+
QUESTION_FILE_PATH = "data/gaia_validation_20.jsonl"
|
8 |
|
9 |
def invoke(level, question, file_name, ground_truth, openai_api_key):
|
10 |
if not question:
|
|
|
13 |
if not openai_api_key:
|
14 |
raise gr.Error("OpenAI API Key is required.")
|
15 |
|
16 |
+
lock = threading.Lock()
|
17 |
+
|
18 |
with lock:
|
19 |
answer = ""
|
20 |
|
21 |
try:
|
22 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
23 |
+
|
24 |
answer = run_crew(question, f"data/{file_name}")
|
25 |
except Exception as e:
|
26 |
raise gr.Error(e)
|
|
|
40 |
outputs=[gr.Textbox(label="Answer", lines=1, interactive=False)],
|
41 |
title="General AI Assistant (GAIA) π€π€π€",
|
42 |
description=os.environ["DESCRIPTION"],
|
43 |
+
examples=get_questions(QUESTION_FILE_PATH),
|
44 |
cache_examples=False
|
45 |
)
|
46 |
|