Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,6 +53,10 @@ def get_similarity(text1, text2, pooling_method, model_id):
|
|
| 53 |
use_mean_pooling = pooling_method == "Use Mean Pooling"
|
| 54 |
embedding1 = get_embedding(text1, use_mean_pooling, model_id)
|
| 55 |
embedding2 = get_embedding(text2, use_mean_pooling, model_id)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
return torch.nn.functional.cosine_similarity(embedding1, embedding2).item()
|
| 57 |
|
| 58 |
|
|
|
|
| 53 |
use_mean_pooling = pooling_method == "Use Mean Pooling"
|
| 54 |
embedding1 = get_embedding(text1, use_mean_pooling, model_id)
|
| 55 |
embedding2 = get_embedding(text2, use_mean_pooling, model_id)
|
| 56 |
+
print("----E1----")
|
| 57 |
+
print(embedding1)
|
| 58 |
+
print("----E2----")
|
| 59 |
+
print(embedding2)
|
| 60 |
return torch.nn.functional.cosine_similarity(embedding1, embedding2).item()
|
| 61 |
|
| 62 |
|