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