iris / app.py
amaanadeen's picture
Upload 2 files
17e1815
raw
history blame contribute delete
635 Bytes
import numpy as np
from flask import Flask, request, jsonify, render_template
import pickle
# Create flask app
flask_app = Flask(__name__)
model = pickle.load(open("model.pkl", "rb"))
@flask_app.route("/")
def Home():
return render_template("index.html")
@flask_app.route("/predict", methods = ["POST"])
def predict():
float_features = [float(x) for x in request.form.values()]
features = [np.array(float_features)]
prediction = model.predict(features)
return render_template("index.html", prediction_text = "The flower species is {}".format(prediction))
if __name__ == "__main__":
flask_app.run(debug=True)