Update app.py

app.py

@@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
 from sklearn.decomposition import PCA
 from sklearn.manifold import TSNE
 model = np.load('gpt2-red-1k-words.npy',allow_pickle='TRUE').item()
-data = np.concatenate([x[0] for x in model.values()], axis=0)
+data = np.asarray([x for x in model.values()])
 keys = [x for x in model.keys()]
 
 cache = "/home/user/app/d.jpg"