Spaces:
Sleeping
Sleeping
felipekitamura
commited on
Update app.py
Browse files
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[
|
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)
|