felipekitamura commited on
Commit
f381ee7
·
verified ·
1 Parent(s): 655b820

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ sp = gr.Image()
64
 
65
 
66
  def inference(word1, word2, word3):
67
- transform = model[word3] - model[word2] + model[word1]
68
  output = model.similar_by_vector(transform)
69
  print(output)
70
  word_list = [word1, word2, word3]
 
64
 
65
 
66
  def inference(word1, word2, word3):
67
+ transform = model[word3] + model[word2] - model[word1]
68
  output = model.similar_by_vector(transform)
69
  print(output)
70
  word_list = [word1, word2, word3]