Spaces:

sonoisa
/

irasuto_search

Sleeping

sonoisa commited on Dec 5, 2021

Commit

5cef811

1 Parent(s): 2e69a1e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -169,7 +169,7 @@ if search_buttion:
     query_embedding = model.encode([query]).numpy()
     distances = scipy.spatial.distance.cdist(
-        [query_embedding], sentence_vectors, metric="cosine"
     )[0]
     results = zip(range(len(distances)), distances)

     query_embedding = model.encode([query]).numpy()
     distances = scipy.spatial.distance.cdist(
+        query_embedding, sentence_vectors, metric="cosine"
     )[0]
     results = zip(range(len(distances)), distances)