Upload app_federal.py

app_federal.py

@@ -290,13 +290,13 @@ def run_retrieval(query):
     query_embeddings = run_dense_retrieval(query)
     #query_embeddings = pca_model.transform(query_embeddings)
     D, I = faiss_index.search(query_embeddings, 100)
-    scores_embeddings = list(D[0])
+    scores_embeddings = list(D[0]) 
     indices_embeddings = I[0]
     indices_embeddings = [int(i) for i in indices_embeddings]
 
     for i in indices_bm25:
         if i not in indices_embeddings:
-            indices_embeddings.append(i)
+            indices_embeddings.append(int(i))
             scores_embeddings.append(-100) #bm25s score is meaningless I think
 
     # ok, and now bm25s as well