felipekitamura commited on
Commit
e3a05a1
·
verified ·
1 Parent(s): 303aac5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def inference(word1, word2, word3):
25
  output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
26
  #out_str = [x + ": " + str(y) for x,y in [item for item in output]]
27
  #return """\n""".join(out_str)
28
- df = pd.DataFrame({"x": [x for x,y in [item for item in output],
29
- "y": [str(y) for x,y in [item for item in output]
30
  })
31
  return df
32
 
 
25
  output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
26
  #out_str = [x + ": " + str(y) for x,y in [item for item in output]]
27
  #return """\n""".join(out_str)
28
+ df = pd.DataFrame({"x": [x for x,y in [item for item in output]],
29
+ "y": [str(y) for x,y in [item for item in output]]
30
  })
31
  return df
32