Update app.py

app.py

@@ -28,6 +28,7 @@ xy = np.c_[xx.ravel(), yy.ravel()]
 y_predicted = knn.predict(xy)
 #plt.pcolormesh(y_predicted.reshape(200, 200), cmap='jet')
 plt.pcolormesh(xx, yy, y_predicted.reshape(200, 200), cmap='jet', alpha=0.2)
+y_unique = np.unique(y_train)
 for i in range(len(y_unique)):
   plt.scatter(x_train[y_train == y_unique[i], 0],
                x_train[y_train == y_unique[i], 1],