Spaces:

ai-law-society-lab
/

Federal_Caselaw_index

Sleeping

dominiks commited on 18 days ago

Commit

9b966cd

verified ·

1 Parent(s): e8cc4b0

Upload app_federal.py

Files changed (1) hide show

app_federal.py CHANGED Viewed

@@ -291,7 +291,7 @@ def run_retrieval(query):
     query_embeddings = run_dense_retrieval(query)
     #query_embeddings = pca_model.transform(query_embeddings)
-    D, I = faiss_index.search(query_embeddings, 45)
     scores_embeddings = list(D[0])
     indices_embeddings = I[0]
     indices_embeddings = [int(i) for i in indices_embeddings]

     query_embeddings = run_dense_retrieval(query)
     #query_embeddings = pca_model.transform(query_embeddings)
+    D, I = faiss_index.search(query_embeddings, 35)
     scores_embeddings = list(D[0])
     indices_embeddings = I[0]
     indices_embeddings = [int(i) for i in indices_embeddings]