Update app.py

app.py

@@ -68,7 +68,7 @@ def inference(word1, word2, word3):
     output = model.similar_by_vector(transform)
     print(output)
     word_list = [word1, word2, word3]
-    word_list.extend([x for x,y in [item for item in output[:4]]])
+    word_list.extend([x for x,y in [item for item in output[:6]]])
     words = {key: model[key] for key in word_list}
     words[word3 + " + (" + word2 + " - " + word1 + ")"] = transform
     data = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)