keithhon commited on
Commit
f179a38
1 Parent(s): e6116c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from sentence_transformers.util import cos_sim
4
  model = SentenceTransformer('keithhon/paraphrase-multilingual-MiniLM-L12-v2')
5
 
6
  def compute_similarity_score(name1, name2):
7
- return cos_sim(model.encode(name1), model.encode(name2))[0]
8
 
9
  iface = gr.Interface(fn=compute_similarity_score, inputs=["text","text"], outputs="text")
10
  iface.launch()
 
4
  model = SentenceTransformer('keithhon/paraphrase-multilingual-MiniLM-L12-v2')
5
 
6
  def compute_similarity_score(name1, name2):
7
+ return cos_sim(model.encode(name1), model.encode(name2)).tolist()[0]
8
 
9
  iface = gr.Interface(fn=compute_similarity_score, inputs=["text","text"], outputs="text")
10
  iface.launch()