-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.py
25 lines (19 loc) · 917 Bytes
/
app.py
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
import pickle
import flask
import pandas as pd
with open(f'model/bike_model_xgboost.pkl','rb') as f:
model=pickle.load(f)
app=flask.Flask(__name__,template_folder='templates')
@app.route('/', methods=['GET','POST'])
def main():
if flask.request.method == 'GET':
return flask.render_template('main.html')
if flask.request.method == 'POST':
temperature=flask.request.form['temperature']
humidity=flask.request.form['humidity']
windspeed=flask.request.form['windspeed']
input_variables=pd.DataFrame([[temperature,humidity,windspeed]],columns=['temperature','humidity','windspeed'],dtype=float)
prediction=model.predict(input_variables)[0]
return flask.render_template('main.html',original_input={'Temperature':temperature,'Humidity':humidity,'Windspeed':windspeed},result=prediction)
if __name__=='__main__':
app.run()