Abdul-Ib commited on
Commit
8ad4f6e
1 Parent(s): 742a705

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -160,14 +160,12 @@ async def predict(query):
160
  semantic_task = asyncio.create_task(semantic_search(query, doc_embeddings))
161
  cheapest_task = asyncio.create_task(is_cheapest(query=query, request_json=product_list))
162
  keyword_scores, semantic_scores, product_list = await asyncio.gather(keyword_task, semantic_task, cheapest_task)
163
- return keyword_scores, semantic_scores, product_list
164
  hybrid_scores = hybrid_search(keyword_scores, semantic_scores)
165
- alter_products_tasks = [process_dict(d) for d in product_list]
166
- product_list = await asyncio.gather(*alter_products_tasks)
167
  results = rerank_results(
168
  request_json=product_list, hybrid_scores=hybrid_scores
169
  )
170
-
 
171
  process_end_time = time.time()
172
  process_time_taken = process_end_time - process_time
173
  time_taken = time.time() - start_time
 
160
  semantic_task = asyncio.create_task(semantic_search(query, doc_embeddings))
161
  cheapest_task = asyncio.create_task(is_cheapest(query=query, request_json=product_list))
162
  keyword_scores, semantic_scores, product_list = await asyncio.gather(keyword_task, semantic_task, cheapest_task)
 
163
  hybrid_scores = hybrid_search(keyword_scores, semantic_scores)
 
 
164
  results = rerank_results(
165
  request_json=product_list, hybrid_scores=hybrid_scores
166
  )
167
+ alter_products_tasks = [process_dict(d) for d in product_list]
168
+ await asyncio.gather(*alter_products_tasks)
169
  process_end_time = time.time()
170
  process_time_taken = process_end_time - process_time
171
  time_taken = time.time() - start_time