felipekitamura commited on
Commit
9d66951
·
verified ·
1 Parent(s): b2748e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ sp = gr.Image()
81
 
82
  def inference(word1, word2, word3):
83
  transform = model[word3] + model[word2] - model[word1]
84
- output = keys[*find_most_similar_vectors(transform[np.newaxis, ...], data)]
85
  print(output)
86
  word_list = [word1, word2, word3]
87
  word_list.extend(output)
 
81
 
82
  def inference(word1, word2, word3):
83
  transform = model[word3] + model[word2] - model[word1]
84
+ output = keys[find_most_similar_vectors(transform[np.newaxis, ...], data)]
85
  print(output)
86
  word_list = [word1, word2, word3]
87
  word_list.extend(output)