felipekitamura commited on
Commit
fbcf726
·
verified ·
1 Parent(s): ff7a342

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ sp = gr.ScatterPlot(x="x", y="y", color="color")
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.expand([x for x,y in output[:4]])
40
  words = {key: model[key] for key in word_list}
41
  data = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)
42
  labels = words.keys()
 
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 output[:4]])
40
  words = {key: model[key] for key in word_list}
41
  data = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)
42
  labels = words.keys()