Update app.py

app.py

@@ -1,10 +1,9 @@
 from flask import Flask, render_template, request
 import joblib
-from sklearn.linear_model import LogisticRegression
 
 app = Flask(__name__)
 
-# Predicting using Saved Model
+# Load the trained model
 model = joblib.load('iris_model.pkl')
 
 @app.route("/")
@@ -13,9 +12,18 @@ def home():
 
 @app.route("/predict", methods=['POST'])
 def predict():
-    pred = model.predict([[6.6,2.8,5.8,1.7]])
-    result = pred[0]
-    return render_template("index.html", data=result)
+    if request.method == 'POST':
+        # Get values from form
+        sl = float(request.form['sl'])
+        sw = float(request.form['sw'])
+        pl = float(request.form['pl'])
+        pw = float(request.form['pw'])
+
+        # Make prediction
+        pred = model.predict([[sl, sw, pl, pw]])
+        result = pred[0]
+
+        return render_template("index.html", data=result)
 
 if __name__ == '__main__':
-    app.run(debug=True, host="0.0.0.0")
+    app.run(debug=True, host="0.0.0.0")