Return csv as json in Flask

Question

I would like to return in JSON the new data forecasts (new_data.csv) of a machine learning model that was saved in joblib.

# -*- coding: utf-8 -*-
from flask import Flask,request, jsonify
from sklearn.externals import joblib
import pandas as pd

app = Flask(__name__)

@app.route('/predict', methods=['GET', 'POST'])

def predict() :    
    json_ = request.json
    new = pd.read_csv('new_data.csv')
    json_vector = new.transform(json_)
    query = pd.DataFrame(json_vector)
    prediction = regr.predict(query)
    return json.dumps({'prediction': list({{prediction}})})

if __name__ == '__main__' :
     regr = joblib.load('model.pkl')
     app.run(port=8080, debug=True)

score 0 · Answer 1 · answered Sep 18 '17 at 14:23

You can use jsonify:

from flask import jsonify

def predict() :    
    json_ = request.json
    new = pd.read_csv('new_data.csv')
    json_vector = new.transform(json_)
    query = pd.DataFrame(json_vector)
    prediction = regr.predict(query)
    data = {'prediction': list({{prediction}})}
    return jsonify(data)

Return csv as json in Flask

1 Answers1