keithhon commited on
Commit
9a9645a
·
1 Parent(s): df46194

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def textToFloatVector(text):
8
  return vec
9
 
10
  def compute(text1, text2):
11
- return cos_sim(textToFloatVector(text1), textToFloatVector(text2))
12
 
13
  iface = gr.Interface(fn=compute, inputs=["text","text"], outputs="text")
14
  iface.launch()
 
8
  return vec
9
 
10
  def compute(text1, text2):
11
+ return cos_sim(textToFloatVector(text1), textToFloatVector(text2)).tolist()[0]
12
 
13
  iface = gr.Interface(fn=compute, inputs=["text","text"], outputs="text")
14
  iface.launch()