keithhon commited on
Commit
773849a
1 Parent(s): 874a120

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -5,7 +5,10 @@ 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]))
 
 
 
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
+ vec1 = model([name1]).numpy().tolist()[0]
9
+ vec2 = model([name2]).numpy().tolist()[0]
10
+ print(vec1)
11
+ return cos_sim(vec1, vec2).numpy().tolist()[0][0]
12
 
13
  iface = gr.Interface(fn=compute_similarity_score, inputs=["text","text"], outputs="text")
14
  iface.launch()