Update app.py

app.py

@@ -38,9 +38,11 @@ def inference(word1, word2, word3):
     word_list = [word1, word2, word3]
     word_list.extend([x for x,y in output[:4]])
     words = {key: model[key] for key in word_list}
-    data = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)
+    data = 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)
     df = pd.DataFrame({
         "x": reduced_data_pca[:, 0], 
         "y": reduced_data_pca[:, 1],