fix

backend/semantic_search.py

@@ -27,7 +27,7 @@ def retrieve(query, k, with_cross_encoder=False):
             scores = cross_encoder.predict([(query, doc[TEXT_COLUMN]) 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 = [elem for elem, _ in sorted_arr[:k]]
+            documents = [documents[index] for _, index in sorted_arr[:k]]
 
         return documents