khalednabawi11 commited on
Commit
27df4b5
·
verified ·
1 Parent(s): ae828fb

Update app/main.py

Browse files
Files changed (1) hide show
  1. 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