everydaytok commited on
Commit
5e1b7f6
·
verified ·
1 Parent(s): e558e99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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