Abdul-Ib commited on
Commit
024b256
1 Parent(s): 9c7a9cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -150,17 +150,18 @@ async def predict(query):
150
  # Perform hybrid search for each query
151
  # this will result in a dictionary of re-ranked search results for each query
152
  process_time = time.time()
153
-
154
- product_list = copy.deepcopy(request_json)
155
- keyword_scores = await check_validity(query, keyword_search)
156
- semantic_scores = await semantic_search(query, doc_embeddings)
157
- hybrid_scores = hybrid_search(keyword_scores, semantic_scores)
158
- product_list = is_cheapest(query=query, request_json=product_list)
159
- for product in product_list:
160
- del product['categoryName'], product['providerName'], product['brandName'], product['productId'], product['key'], product['productOldPrice'], product['imageUrl'], product['currency'], product['providerLogo'], product['productUrl'], product['productRatingCount'], product['productRating'], product['productType']
161
- results = rerank_results(
162
- request_json=product_list, hybrid_scores=hybrid_scores
163
- )
 
164
 
165
  process_end_time = time.time()
166
  process_time_taken = process_end_time - process_time
 
150
  # Perform hybrid search for each query
151
  # this will result in a dictionary of re-ranked search results for each query
152
  process_time = time.time()
153
+
154
+ for query in normalized_query_list:
155
+ product_list = copy.deepcopy(request_json)
156
+ keyword_scores = await check_validity(query, keyword_search)
157
+ semantic_scores = await semantic_search(query, doc_embeddings)
158
+ hybrid_scores = hybrid_search(keyword_scores, semantic_scores)
159
+ product_list = is_cheapest(query=query, request_json=product_list)
160
+ for product in product_list:
161
+ del product['categoryName'], product['providerName'], product['brandName'], product['productId'], product['key'], product['productOldPrice'], product['imageUrl'], product['currency'], product['providerLogo'], product['productUrl'], product['productRatingCount'], product['productRating'], product['productType']
162
+ results = rerank_results(
163
+ request_json=product_list, hybrid_scores=hybrid_scores
164
+ )
165
 
166
  process_end_time = time.time()
167
  process_time_taken = process_end_time - process_time