Spaces:
Configuration error
Configuration error
File size: 2,043 Bytes
7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf ab8cdf3 7b39dbf |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from flask import Flask, render_template, request
import os
import numpy as np
import pandas as pd
from mlProject.pipeline.stage06_prediction import PredictionPipeline
app=Flask(__name__)
@app.route('/',methods=['GET'])
def homepage():
return render_template("index.html")
@app.route('/train',methods=['GET'])
def training():
os.system("python main.py")
return "Training Successful"
@app.route('/predict',methods=['POST','GET']) # route to show the predictions in a web UI
def index():
if request.method == 'POST':
try:
# reading the inputs given by the user
CustomerId =int(request.form['CustomerId'])
CreditScore =int(request.form['CreditScore'])
Gender =int(request.form['Gender'])
Age =int(request.form['Age'])
Tenure =int(request.form['Tenure'])
Balance =float(request.form['Balance'])
NumOfProducts =int(request.form['NumOfProducts'])
HasCrCard =int(request.form['HasCrCard'])
IsActiveMember=int(request.form['IsActiveMember'])
EstimatedSalary =float(request.form['EstimatedSalary'])
Geography_Germany =int(request.form['Geography_Germany'])
Geography_Spain =int(request.form['Geography_Spain'])
data = [CustomerId,CreditScore,Gender,Age,Tenure,Balance,NumOfProducts,HasCrCard,IsActiveMember,EstimatedSalary,Geography_Germany,Geography_Spain]
data = np.array(data).reshape(1, 12)
obj = PredictionPipeline()
predict = obj.predict(data)
result = "Likely to Churn" if predict == 1 else "Unlikely to Churn"
return result
except Exception as e:
print('The Exception message is: ',e)
return 'something is wrong'
else:
return render_template('index.html')
# Another trial
if __name__=="__main__":
# app.run(host="0.0.0.0", port=8080, debug=True)
app.run(host="0.0.0.0", port=8080)
|