GPalomeque
commited on
Commit
•
53d0b05
1
Parent(s):
c71280c
Update app.py
Browse files
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(
|
117 |
-
embeddings2 = model.encode(
|
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)
|