Spaces:
Runtime error
Runtime error
Update app/main.py
Browse files- app/main.py +5 -1
app/main.py
CHANGED
@@ -411,8 +411,12 @@ async def ask(query: Query):
|
|
411 |
prompt = generate_prompt(query.question)
|
412 |
timeout_callback = TimeoutCallback(timeout_seconds=60)
|
413 |
|
|
|
|
|
|
|
414 |
answer = await asyncio.wait_for(
|
415 |
-
qa_chain.run(prompt, callbacks=[timeout_callback]),
|
|
|
416 |
timeout=60
|
417 |
)
|
418 |
|
|
|
411 |
prompt = generate_prompt(query.question)
|
412 |
timeout_callback = TimeoutCallback(timeout_seconds=60)
|
413 |
|
414 |
+
|
415 |
+
loop = asyncio.get_event_loop()
|
416 |
+
|
417 |
answer = await asyncio.wait_for(
|
418 |
+
# qa_chain.run(prompt, callbacks=[timeout_callback]),
|
419 |
+
loop.run_in_executor(None, qa_chain.run, prompt),
|
420 |
timeout=60
|
421 |
)
|
422 |
|