Tonic commited on
Commit
1902a05
1 Parent(s): 99939d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ def compute_similarity(selected_task, sentence1, sentence2, extra_sentence1, ext
99
  # Free memory
100
  free_memory(embeddings1, embeddings2, embeddings3, embeddings4)
101
 
102
- return similarity1, similarity2, similarity3
103
 
104
  @spaces.GPU
105
  def compute_cosine_similarity(emb1, emb2):
 
99
  # Free memory
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):