Spaces:
Runtime error
Runtime error
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}")
|