Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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()
|