Spaces:

eagle0504
/

document-search-q-series

Running

eagle0504 commited on Apr 22, 2024

Commit

e4026af

verified ·

1 Parent(s): 0675d90

Update helper/utils.py

Files changed (1) hide show

helper/utils.py CHANGED Viewed

@@ -251,7 +251,7 @@ def query_search(
     # Rename columns for clarity
     refs = refs.rename(
         # columns={0: "sentences", 1: "query_embeddings", 2: "page no", 3: "qim"}
-        columns={0: "sentences", 1: "page no", 2: "score"}
     )
     # Sort the DataFrame based on the 'qim' score in descending order
     refs = refs.sort_values(by="qim", ascending=False)

     # Rename columns for clarity
     refs = refs.rename(
         # columns={0: "sentences", 1: "query_embeddings", 2: "page no", 3: "qim"}
+        columns={0: "sentences", 1: "page no", 2: "qim"}
     )
     # Sort the DataFrame based on the 'qim' score in descending order
     refs = refs.sort_values(by="qim", ascending=False)