ML_RFA / main.py
LazyBoss's picture
Update main.py
6c94057 verified
import numpy as np
from flask import Flask, request, render_template
import pickle
app = Flask(__name__)
model = pickle.load(open('model.pkl', 'rb'))
@app.route('/')
def home():
return render_template('index.html')
@app.route('/predict', methods=['POST', 'GET'])
def predict():
int_features = [float(x) for x in request.form.values()]
final_features = [np.array(int_features)]
prediction = model.predict(final_features)
if prediction == 0:
return 'Low chances of transaction being fraud'
else:
return 'High chances of transaction being fraud'
if __name__ == "__main__":
app.run(debug=True)