Update app.py

app.py

@@ -87,12 +87,12 @@ def inference(word1, word2, word3):
     word_list.extend(output)
     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)
+    data2 = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)
     #print(data.shape)
     labels = words.keys()
     #reduced_data_pca = reduce_dimensions(data, method='PCA')
     #print(reduced_data_pca.shape)
-    plot_reduced_data(data, labels, 'PCA Results')
+    plot_reduced_data(data2, labels, 'PCA Results')
     return cache
 
 examples = [