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()
|