Update app.py

app.py

@@ -155,45 +155,7 @@ def search(query):
         )
         
 
-    #### Sematic Search #####
-    # Encode the query using the bi-encoder and find potentially relevant passages
-    question_embedding = bi_encoder.encode(query, convert_to_tensor=True)
-#     question_embedding = question_embedding.cuda()
-    hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=top_k)
-    hits = hits[0]  # Get the hits for the first query
-
-    ##### Re-Ranking #####
-    # Now, score all retrieved passages with the cross_encoder
-    cross_inp = [[query, passages[hit['corpus_id']]] for hit in hits]
-    cross_scores = cross_encoder.predict(cross_inp)
-
-    # Sort results by the cross-encoder scores
-    for idx in range(len(cross_scores)):
-        hits[idx]['cross-score'] = cross_scores[idx]
-
-    # Output of top-5 hits from bi-encoder
-    print("\n-------------------------\n")
-    print("Top-5 Bi-Encoder Retrieval hits")
-    hits = sorted(hits, key=lambda x: x['score'], reverse=True)
-    for hit in hits[0:5]:
-#         print("\t{:.3f}\t{}".format(hit['score'], passages[hit['corpus_id']].replace("\n", " ")))
-        row_dict = df.loc[df['all_review']== corpus[hit['corpus_id']]]
-        print("\t{:.3f}\t".format(hit['score']),row_dict['Hotel'].values[0])
-        de = df_basic.loc[df_basic.Hotel == row_dict['Hotel'].values[0]]
-        print(f'\tPrice Per night: {de.price_per_night.values[0]}')
-        print(de.description.values[0])
-
-    # Output of top-5 hits from re-ranker
-    print("\n-------------------------\n")
-    print("Top-5 Cross-Encoder Re-ranker hits")
-    hits = sorted(hits, key=lambda x: x['cross-score'], reverse=True)
-    for hit in hits[0:5]:
-#         print("\t{:.3f}\t{}".format(hit['cross-score'], passages[hit['corpus_id']].replace("\n", " ")))
-        row_dict = df.loc[df['all_review']== corpus[hit['corpus_id']]]
-        print("\t{:.3f}\t".format(hit['cross-score']),row_dict['Hotel'].values[0])
-        de = df_basic.loc[df_basic.Hotel == row_dict['Hotel'].values[0]]
-        print(f'\tPrice Per night: {de.price_per_night.values[0]}')
-        print(de.description.values[0])
+    
         
         
     return bm25list