asigalov61 commited on
Commit
08fa436
·
verified ·
1 Parent(s): 7c5baac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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[:, :2048], embeddings[:, :2048])
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