keithhon commited on
Commit
566338a
1 Parent(s): b50097e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,10 +1,11 @@
1
- import gradio as gr
2
- from sentence_transformers import SentenceTransformer
3
  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)).tolist()[0]
8
 
9
  iface = gr.Interface(fn=compute_similarity_score, inputs=["text","text"], outputs="text")
10
  iface.launch()
 
 
 
1
  from sentence_transformers.util import cos_sim
2
+ import gradio as gr
3
+ module_url = "https://tfhub.dev/google/universal-sentence-encoder/4"
4
+ import tensorflow_hub as hub
5
+ model = hub.load(module_url)
6
 
7
  def compute_similarity_score(name1, name2):
8
+ return cos_sim(model([name1]), model([name2])).tolist()[0]
9
 
10
  iface = gr.Interface(fn=compute_similarity_score, inputs=["text","text"], outputs="text")
11
  iface.launch()