Update without Cross-encoder model

app.py

@@ -40,11 +40,11 @@ class DocumentSearch:
         # perform search via faiss FlatIP index
         distances, indeces = self.index.search(query_vector, k*10)
         # get docs by index
-        docs = [self.labels[i] for i in indeces[0]]
+        res_docs = [self.docs[i] for i in indeces[0]]
         # get scores by index
         dists = [dist for dist in distances[0]]
         
-        return[{'doc': doc[0], 'url': doc[1], 'score': dist} for doc, dist in zip(docs, dists)]
+        return[{'doc': doc[0], 'url': doc[1], 'score': dist} for doc, dist in zip(res_docs, dists)]
         ##### OLD VERSION WITH CROSS-ENCODER #####
         # get answers by index
         #answers = [self.docs[i] for i in indeces[0]]