Spaces:

nickmuchi
/

semantic-search-with-retrieve-and-rerank

Running

nickmuchi commited on Aug 9, 2023

Commit

83a2deb

1 Parent(s): 11fdeb2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,7 @@ def bi_encode(bi_enc,passages):
             instruction = "Represent this sentence for searching relevant passages: "
-            corpus_embeddings = bi_encoder.encode([instruction + sentence for sentence in passages], convert_to_tensor=True)
         else:

             instruction = "Represent this sentence for searching relevant passages: "
+            corpus_embeddings = bi_encoder.encode([instruction + sentence for sentence in passages], normalize_embeddings=True)
         else: