-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
28 lines (24 loc) · 776 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
26
27
28
import pickle
from flask import Flask, request, app, jsonify, url_for, render_template
from django.shortcuts import render
import numpy as np
import pandas as pd
from flask import Flask
app=Flask(__name__)
#load the model
pickled_model=pickle.load(open('rfrmodel.pkl','rb'))
scalar=pickle.load(open('scaling.pkl','rb'))
@app.route('/')
def home():
return render_template('home.html')
@app.route('/predict_api',methods=['POST'])
def predict_api():
data=request.json['data']
print(data)
print(np.array(list(data.values())).reshape(1,-1))
new_data=scalar.transform((np.array(list(data.values())).reshape(1,-1)))
output=pickled_model.predict(new_data)
print(output[0])
return jsonify(output[0])
if __name__=="__main__":
app.run(debug=True)