bikram-riverus commited on
Commit
dc111b4
·
verified ·
1 Parent(s): 2449492
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ model = SentenceTransformer("rufimelo/Legal-BERTimbau-sts-base")
9
 
10
  def compute_similarity(left_text: str, right_text: str) -> np.ndarray:
11
  embeddings = model.encode([left_text, right_text])
 
12
  similarity = cosine_similarity(embeddings[: 1], embeddings[1 :])
13
 
14
  return similarity
 
9
 
10
  def compute_similarity(left_text: str, right_text: str) -> np.ndarray:
11
  embeddings = model.encode([left_text, right_text])
12
+ print(embeddings)
13
  similarity = cosine_similarity(embeddings[: 1], embeddings[1 :])
14
 
15
  return similarity