Skip to content

Instantly share code, notes, and snippets.

@harsha89
Last active May 26, 2020 13:32
Show Gist options
  • Save harsha89/56fee39ff34d42392035c203f729f6df to your computer and use it in GitHub Desktop.
Save harsha89/56fee39ff34d42392035c203f729f6df to your computer and use it in GitHub Desktop.
from flask import Flask, jsonify, request
import pandas as pd
import joblib
app = Flask(__name__)
@app.route("/predict", methods=['POST'])
def do_prediction():
json = request.get_json()
model = joblib.load('model/rf_model.pkl')
df = pd.DataFrame(json, index=[0])
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()
scaler.fit(df)
df_x_scaled = scaler.transform(df)
df_x_scaled = pd.DataFrame(df_x_scaled, columns=df.columns)
y_predict = model.predict(df_x_scaled)
result = {"Predicted House Price" : y_predict[0]}
return jsonify(result)
if __name__ == "__main__":
app.run(host='0.0.0.0')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment