fix

app.py

@@ -34,7 +34,7 @@ def add_text(history, text):
     return history, gr.Textbox(value="", interactive=False)
 
 
-def bot(history, api_kind, with_cross_encoder):
+def bot(history, api_kind, with_cross_encoder=False):
     query = history[-1][0]
 
     if not query:

backend/semantic_search.py

@@ -24,7 +24,8 @@ def retrieve(query, k, with_cross_encoder=False):
             documents = [doc[TEXT_COLUMN] for doc in documents]
         else:
             documents = TABLE.search(query_vec, vector_column_name=VECTOR_COLUMN).limit(k * 2).to_list()
-            scores = cross_encoder.predict([(query, doc[TEXT_COLUMN]) for doc in documents])
+            documents = [doc[TEXT_COLUMN] for doc in documents]
+            scores = cross_encoder.predict([(query, doc) for doc in documents])
             indexed_arr = [(elem, index) for index, elem in enumerate(scores)]
             sorted_arr = sorted(indexed_arr, key=lambda x: x[0], reverse=True)
             documents = [documents[index] for _, index in sorted_arr[:k]]