Update app.py

app.py

@@ -94,8 +94,14 @@ def predict():
 
     res_idx = iris_model.predict([[sl, sw, pl, pw]])[0]
     res = species[res_idx]
-    flash(f"The predicted flower species is: {res}", "success")
-    return redirect(url_for("dashboard"))
+    # flash(f"The predicted flower species is: {res}", "success")
+    # return redirect(url_for("dashboard"))
+        # Render dashboard with prediction
+    return render_template(
+        "dashboard.html",
+        username=session["user"],
+        prediction=res
+    )
 
 @app.route("/portfolio")
 def portfolio():