File size: 919 Bytes
ba6b28a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import pickle
from flask import Flask, render_template, request
IRIS_TYPES = {
0: 'setosa',
1: 'versicolor',
2: 'virginica'
}
app = Flask(__name__)
@app.route('/', methods=['GET'])
def iris_index():
return render_template('index.html')
@app.route('/predict/', methods=['POST'])
def result():
if request.method == 'POST':
sepal_length = request.form['inputSepalLength']
sepal_width = request.form['inputSepalWidth']
petal_length = request.form['inputPetalLength']
petal_width = request.form['inputPetalWidth']
data = [[sepal_length, sepal_width, petal_length, petal_width]]
model = pickle.load(open(f"model.sav", 'rb'))
pred = model.predict(data)[0]
return render_template('prediction.html', iris_type=IRIS_TYPES[pred])
if __name__ == '__main__':
app.debug = True
app.run(
host='0.0.0.0',
port=7860,
debug=True)
|