Update app.py

app.py

@@ -2,7 +2,7 @@ from flask import Flask, render_template, request, redirect, url_for, session, f
 from werkzeug.security import generate_password_hash, check_password_hash
 
 # ML imports
-import seaborn as sns
+from sklearn.datasets import load_iris
 from sklearn.linear_model import LogisticRegression
 import numpy as np
 
@@ -12,9 +12,10 @@ app.secret_key = "supersecretkey"  # ⚠️ change this in production!
 # ----------------------------
 # Train Iris Model (once only)
 # ----------------------------
-iris = sns.load_dataset("iris")
-X = iris.iloc[:, 0:4].values
-y = iris.iloc[:, 4].values
+iris = load_iris(as_frame=True)
+X = iris.data.values
+y = iris.target
+species = iris.target_names
 
 iris_model = LogisticRegression(max_iter=300)
 iris_model.fit(X, y)
@@ -85,7 +86,8 @@ def predict():
         flash("Invalid input values!", "danger")
         return redirect(url_for("dashboard"))
 
-    res = iris_model.predict([[sl, sw, pl, pw]])[0]
+    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"))