| import joblib
|
| from flask import Flask,render_template,request
|
| import numpy as np
|
|
|
| app = Flask(__name__)
|
|
|
| MODEL_PATH = "artifacts/models/model.pkl"
|
| model = joblib.load(MODEL_PATH)
|
|
|
|
|
| FEATURES = [
|
| 'Location', 'MinTemp', 'MaxTemp', 'Rainfall', 'Evaporation', 'Sunshine',
|
| 'WindGustDir', 'WindGustSpeed', 'WindDir9am', 'WindDir3pm',
|
| 'WindSpeed9am', 'WindSpeed3pm', 'Humidity9am', 'Humidity3pm',
|
| 'Pressure9am', 'Pressure3pm', 'Cloud9am', 'Cloud3pm', 'Temp9am',
|
| 'Temp3pm', 'RainToday', 'Year', 'Month', 'Day'
|
| ]
|
|
|
| LABELS = {0 : "NO" , 1: "YES"}
|
|
|
| @app.route("/" , methods=["GET" , "POST"])
|
| def index():
|
| prediction = None
|
|
|
| if request.method=="POST":
|
| try:
|
| input_data = [float(request.form[feature]) for feature in FEATURES]
|
| input_array = np.array(input_data).reshape(1,-1)
|
|
|
| pred = model.predict(input_array)[0]
|
| prediction = LABELS.get(pred, 'Unknown')
|
|
|
|
|
| except Exception as e:
|
| print(str(e))
|
|
|
| return render_template("index.html" , prediction=prediction , features=FEATURES)
|
|
|
| if __name__=="__main__":
|
| app.run(debug=True , port=5000 , host="0.0.0.0")
|
|
|
