felipekitamura commited on
Commit
ceb61af
·
verified ·
1 Parent(s): dc72e07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ sp = gr.ScatterPlot(x="x", y="y", color="color")
35
 
36
  def inference(word1, word2, word3):
37
  output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
 
38
  word_list = [word1, word2, word3]
39
  word_list.extend([x for x,y in [item for item in output[:4]]])
40
  words = {key: model[key] for key in word_list}
 
35
 
36
  def inference(word1, word2, word3):
37
  output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
38
+ print(output)
39
  word_list = [word1, word2, word3]
40
  word_list.extend([x for x,y in [item for item in output[:4]]])
41
  words = {key: model[key] for key in word_list}