Spaces:
Running
Running
felipekitamura
commited on
Update app.py
Browse files
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}
|