Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
5 |
|
6 |
def compute_similarity_score(name1, name2):
|
7 |
-
return cos_sim(model
|
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()
|