Upload app_federal.py

app_federal.py

@@ -205,6 +205,7 @@ def run_retrieval(query):
     D, I = faiss_index.search(query_embeddings, 45)
     scores_embeddings = D[0]
     indices_embeddings = I[0]
+    indices_embeddings = [int(i) for i in indices_embeddings]
 
     #results = [{"index":i, "NV_score":j, "text": chunks[i]} for i,j in zip(indices_embeddings, scores_embeddings)]
     results = [{"id":i, "NV_score":j, "text":ds_paragraphs[i]["paragraph"]} for i,j in zip(indices_embeddings, scores_embeddings)]