Flask model API

flask_model_api.py

from flask import request, Flask
import json
import pickle

app = Flask(__name__)


@app.route('/', methods=['GET', 'POST'])
def main_flask_function():
    if request.method == 'GET':
        return 'Use post method'
    else:
        req_data = json.loads(request.data)
        print(list(req_data.values()))
        with open('clf_wine.pickle', 'rb') as f:
            clf_new = pickle.load(f)

            print( list( map(int, clf_new.predict( [list(req_data.values())] ) ) ) )
            
            return {'prediction':list(map(int,clf_new.predict([list(req_data.values())])))} 


if __name__ == '__main__':
    app.run(host = '0.0.0.0',debug=True)