bikram-riverus commited on
Commit
2449492
·
verified ·
1 Parent(s): c2461a6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +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[0], embeddings[1])
13
 
14
  return similarity
15
 
 
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
15