Update app.py
Browse files
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()
|