Tonic commited on
Commit
4cd22b7
·
verified ·
1 Parent(s): 36cbf30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -100,7 +100,8 @@ def compute_similarity(selected_task, sentence1, sentence2, extra_sentence1, ext
100
  free_memory(embeddings1, embeddings2, embeddings3, embeddings4)
101
 
102
  similarity_scores = {"Similarity 1-2": similarity1, "Similarity 1-3": similarity2, "Similarity 1-4": similarity3}
103
-
 
104
  # @spaces.GPU
105
  def compute_cosine_similarity(emb1, emb2):
106
  tensor1 = torch.tensor(emb1).to(device).half()
 
100
  free_memory(embeddings1, embeddings2, embeddings3, embeddings4)
101
 
102
  similarity_scores = {"Similarity 1-2": similarity1, "Similarity 1-3": similarity2, "Similarity 1-4": similarity3}
103
+ return similarity_scores
104
+
105
  # @spaces.GPU
106
  def compute_cosine_similarity(emb1, emb2):
107
  tensor1 = torch.tensor(emb1).to(device).half()