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