semantic-search-with-retrieve-and-rerank

Sleeping

nickmuchi commited on May 27, 2023

Commit

61c36ad

•

1 Parent(s): a4f7662

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -138,7 +138,7 @@ def bi_encode(bi_enc,passages):
     #Compute the embeddings using the multi-process pool
     with st.spinner('Encoding passages into a vector space...'):
-        if bi_enc == 'intfloat/e5-base':
             corpus_embeddings = bi_encoder.encode(['passage: ' + sentence for sentence in passages], convert_to_tensor=True)

     #Compute the embeddings using the multi-process pool
     with st.spinner('Encoding passages into a vector space...'):
+        if bi_enc == 'intfloat/e5-base-v2':
             corpus_embeddings = bi_encoder.encode(['passage: ' + sentence for sentence in passages], convert_to_tensor=True)