Update app.py

app.py

@@ -168,10 +168,10 @@ def bm25_api(passages):
 
 bi_enc_options = ["multi-qa-mpnet-base-dot-v1","all-mpnet-base-v2","multi-qa-MiniLM-L6-cos-v1"]
 
-def display_df_as_table(model,top_k,score):
+def display_df_as_table(model,top_k,score='score'):
     # Display the df with text and scores as a table
     df = pd.DataFrame([(hit[score],passages[hit['corpus_id']]) for hit in model[0:top_k]],columns=['Score','Text'])
-    df['Score'] = round(df['Score'],3)
+    df['Score'] = round(df['Score'],2)
     
     return df
         
@@ -211,7 +211,7 @@ def search_func(query, top_k=top_k):
     
     st.subheader(f"Top-{top_k} lexical search (BM25) hits")
     
-    bm25_df = display_df_as_table(bm25_hits,top_k,'score')
+    bm25_df = display_df_as_table(bm25_hits,top_k)
     st.write(bm25_df.to_html(index=False), unsafe_allow_html=True)
 
     ##### Sematic Search #####
@@ -235,7 +235,7 @@ def search_func(query, top_k=top_k):
     st.subheader(f"Top-{top_k} Bi-Encoder Retrieval hits")
     hits = sorted(hits, key=lambda x: x['score'], reverse=True)
     
-    cross_df = display_df_as_table(hits,top_k,'score')
+    cross_df = display_df_as_table(hits,top_k)
     st.write(cross_df.to_html(index=False), unsafe_allow_html=True)
 
     # Output of top-3 hits from re-ranker