Update app.py
Browse files
app.py
CHANGED
@@ -144,7 +144,7 @@ def bi_encode(bi_enc,passages):
|
|
144 |
|
145 |
else:
|
146 |
|
147 |
-
corpus_embeddings = bi_encoder.encode(
|
148 |
|
149 |
|
150 |
st.success(f"Embeddings computed. Shape: {corpus_embeddings.shape}")
|
|
|
144 |
|
145 |
else:
|
146 |
|
147 |
+
corpus_embeddings = bi_encoder.encode(passages, convert_to_tensor=True)
|
148 |
|
149 |
|
150 |
st.success(f"Embeddings computed. Shape: {corpus_embeddings.shape}")
|