Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -235,7 +235,7 @@ def select_best_output(outputs, embeddings, src_embeddings, input_mixed_pooling,
|
|
| 235 |
emb_sims = cosine_similarity_numpy(src_embeddings, embeddings)
|
| 236 |
|
| 237 |
else:
|
| 238 |
-
emb_sims = cosine_similarity_numpy(src_embeddings[
|
| 239 |
|
| 240 |
sorted_emb_sims = sorted(emb_sims, reverse=True)
|
| 241 |
|
|
|
|
| 235 |
emb_sims = cosine_similarity_numpy(src_embeddings, embeddings)
|
| 236 |
|
| 237 |
else:
|
| 238 |
+
emb_sims = cosine_similarity_numpy(src_embeddings[:2048], embeddings[:, :2048])
|
| 239 |
|
| 240 |
sorted_emb_sims = sorted(emb_sims, reverse=True)
|
| 241 |
|