Update app.py

app.py

@@ -28,9 +28,11 @@ def predict(alpha):
   ridge_reg = build_model(alpha)
   preds = ridge_reg.predict(X_test)
   fig = plt.figure()
-  plt.plot(X_train, y_train, "g-", label="train data")
-  plt.plot(X_test, y_test, "r-", label="test data")
-  plt.plot(X_test, preds, "b--", label="regularized model")
+  plt.scatter(X_train, y_train, color="yellowgreen", marker=".", label="training data")
+  plt.scatter(X_test, y_test, color="gold", marker=".", label="test data")
+  plt.plot(X_test, preds, color="cornflowerblue",
+    linewidth=2,
+    label="Ridge  regressor")
   plt.ylabel("Y")
   plt.xlabel("X")
   return plt