felipekitamura commited on
Commit
66642ba
·
verified ·
1 Parent(s): 410ea05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -20,7 +20,9 @@ Word3 = gr.Textbox()
20
  label = gr.Label(show_label=True, label="Word4")
21
 
22
  def inference(word1, word2, word3):
23
- return '\n'.join(model.similar_by_vector(model[word3] + model[word2] - model[word1]))
 
 
24
 
25
  examples = [
26
  ["woman", "man", "aunt"],
 
20
  label = gr.Label(show_label=True, label="Word4")
21
 
22
  def inference(word1, word2, word3):
23
+ output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
24
+ out_str = [x + ": " + str(y) + "\n" for x,y in [item for item in output]]
25
+ return '\n'.join()
26
 
27
  examples = [
28
  ["woman", "man", "aunt"],