Update app.py

app.py

@@ -53,10 +53,11 @@ def getfig(X_test):
 
     fig = plt.figure(figsize=(5,4))
 
-    plt.scatter(X_pca[:, 0], X_pca[:, 1], c = y, cmap = plt.cm.Spectral, s =  10)
+    plt.scatter(X_pca[:, 0], X_pca[:, 1], c = y, cmap = plt.cm.Spectral, s =  10, label=['More HA','Less HA'])
     plt.scatter(u_pca[:, 0], u_pca[:, 1], c = 'g', cmap = plt.cm.Spectral, s =  40)
-    plt.annotate('You', (u_pca[:, 0]+10, u_pca[:, 1]+10), bbox = dict(boxstyle="round",ec='g',fc='lightgreen'))
+    plt.annotate('You', (u_pca[:, 0]+5, u_pca[:, 1]+5), bbox = dict(boxstyle="round",ec='g',fc='lightgreen'))
     plt.title(f"PCA, Exp. Variance: {np.round(np.sum(pca.explained_variance_ratio_), 4)}")
+    plt.legend()
     plt.xlabel("PC 1")
     plt.ylabel("PC 2")