Update app.py

app.py

@@ -38,10 +38,10 @@ if st.button('Compute similarity'):
     print('Generating visualization...')
     sentembs, data = load_plot_data()
     X_embedded = TSNE(n_components=2, learning_rate='auto',
-                  init='random').fit_transform(np.concatenate([sentembs, encodings], axis=1))
+                  init='random').fit_transform(np.concatenate([sentembs, encodings], axis=0))
                   
-    data.append({'sent': sent1, 'color': '#F0E442'}) # sentence 1
-    data.append({'sent': sent2, 'color': '#D55E00'}) # sentence 2
+    data = data.append({'sent': sent1, 'color': '#F0E442'}) # sentence 1
+    data = data.append({'sent': sent2, 'color': '#D55E00'}) # sentence 2
     data['x'] = X_embedded[:,0]
     data['y'] = X_embedded[:,1]