Spaces:

Penality
/

pdf-something

Sleeping

Penality commited on Feb 23

Commit

a06984f

verified ·

1 Parent(s): 2270bc7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def retrieve_document(query):
     query_embedding = embedding_model.encode([query])
     _, closest_idx = index.search(np.array(query_embedding, dtype=np.float32), 1)
-    print(f"retrieved: {documents[closest_idx[0][0]}")
     return documents[closest_idx[0][0]]

     query_embedding = embedding_model.encode([query])
     _, closest_idx = index.search(np.array(query_embedding, dtype=np.float32), 1)
+    print(f"retrieved: {documents[closest_idx[0][0]]}")
     return documents[closest_idx[0][0]]