Tonic commited on
Commit
2d4bf4a
Β·
verified Β·
1 Parent(s): 3331ada

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -101,6 +101,9 @@ def compute_similarity(selected_task, sentence1, sentence2, extra_sentence1, ext
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()
 
101
 
102
  similarity_scores = {"Similarity 1-2": similarity1, "Similarity 1-3": similarity2, "Similarity 1-4": similarity3}
103
 
104
+ return similarity_scores
105
+
106
+
107
  @spaces.GPU
108
  def compute_cosine_similarity(emb1, emb2):
109
  tensor1 = torch.tensor(emb1).to(device).half()