Spaces:

singhvaibhav924
/

Research_Bot

Sleeping

singhvaibhav924 commited on Jul 18, 2024

Commit

ce136a3

1 Parent(s): e20f6e0

minor bug fix

Files changed (1) hide show

helper.py CHANGED Viewed

@@ -66,7 +66,7 @@ def re_rank_papers(model, query_abstract, papers):
     target_embeddings = model.encode([query_abstract])
     summaries_embeddings = model.encode(list(summaries.keys()))
-    cosine_similarities = -torch.nn.functional.cosine_similarity(target_embeddings, summaries_embeddings)
     cosine_similarities = cosine_similarities.tolist()
     i = 0

     target_embeddings = model.encode([query_abstract])
     summaries_embeddings = model.encode(list(summaries.keys()))
+    cosine_similarities = -torch.nn.functional.cosine_similarity(torch.from_numpy(target_embeddings), torch.from_numpy(summaries_embeddings))
     cosine_similarities = cosine_similarities.tolist()
     i = 0