testiris / app.py
LovnishVerma's picture
Update app.py
1154d44 verified
raw
history blame contribute delete
730 Bytes
from flask import Flask, render_template, request
import joblib
app = Flask(__name__)
# Load the trained model
model = joblib.load('iris_model (3).pkl')
@app.route("/")
def home():
return render_template("index.html")
@app.route("/predict", methods=['POST'])
def predict():
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")