Update app.py

app.py

@@ -14,7 +14,7 @@ def reduce_dimensions(data, method='PCA'):
     if method == 'PCA':
         model = PCA(n_components=2)
     elif method == 'TSNE':
-        model = TSNE(n_components=2, learning_rate='auto', init='random', perplexity=3)
+        model = TSNE(n_components=2, learning_rate='auto', init='random', perplexity=4)
     return model.fit_transform(data)
 
 # Plotting function
@@ -74,9 +74,9 @@ def inference(word1, word2, word3):
     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')
+    reduced_data_pca = reduce_dimensions(data, method='TSNE')
     print(reduced_data_pca.shape)
-    plot_reduced_data(reduced_data_pca, labels, 'PCA Results')
+    plot_reduced_data(reduced_data_pca, labels, 'TSNE Results')
     return cache
 
 examples = [