-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathmain.py
31 lines (24 loc) · 917 Bytes
/
main.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
26
27
28
29
30
31
from flask import Flask, request, render_template
import pickle
app = Flask(__name__)
#importing pickle files
model = pickle.load(open('classifier.pkl','rb'))
ferti = pickle.load(open('fertilizer.pkl','rb'))
@app.route('/')
def welcome():
return render_template('index.html')
@app.route('/predict',methods=['POST'])
def predict():
temp = request.form.get('temp')
humi = request.form.get('humid')
mois = request.form.get('mois')
soil = request.form.get('soil')
crop = request.form.get('crop')
nitro = request.form.get('nitro')
pota = request.form.get('pota')
phosp = request.form.get('phos')
input = [int(temp),int(humi),int(mois),int(soil),int(crop),int(nitro),int(pota),int(phosp)]
res = ferti.classes_[model.predict([input])]
return render_template('index.html',x = ('Predicted Fertilizer is {}'.format(res)))
if __name__ == "__main__":
app.run(debug=True)