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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ 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()
 
5
  model = hub.load(module_url)
6
 
7
  def compute_similarity_score(name1, name2):
8
+ return cos_sim(model([name1]), model([name2])).tolist()
9
 
10
  iface = gr.Interface(fn=compute_similarity_score, inputs=["text","text"], outputs="text")
11
  iface.launch()