GPalomeque commited on
Commit
53d0b05
1 Parent(s): c71280c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -113,8 +113,8 @@ def clasifica_conv_americana(example):
113
  def similitud(example,example2):
114
  model = SentenceTransformer("hackathon-pln-es/jurisbert-tsdae-sentence-transformer")
115
  #Compute embedding for both lists
116
- embeddings1 = model.encode(sentences1, convert_to_tensor=True)
117
- embeddings2 = model.encode(sentences2, convert_to_tensor=True)
118
 
119
  #Compute cosine-similarits
120
  cosine_scores = util.cos_sim(embeddings1, embeddings2)
 
113
  def similitud(example,example2):
114
  model = SentenceTransformer("hackathon-pln-es/jurisbert-tsdae-sentence-transformer")
115
  #Compute embedding for both lists
116
+ embeddings1 = model.encode(example, convert_to_tensor=True)
117
+ embeddings2 = model.encode(example2, convert_to_tensor=True)
118
 
119
  #Compute cosine-similarits
120
  cosine_scores = util.cos_sim(embeddings1, embeddings2)