import numpy as np from flask import Flask, request, jsonify import pickle app = Flask(__name__) model = pickle.load(open('gpt-model.pkl','rb')) @app.route('/api',methods=['POST']) def completions(): data = request.get_json(force=True) prediction = model.predict([[np.array(data['exp'])]]) output = prediction[0] return jsonify(output) if __name__ == '__main__': app.run(port=5000, debug=True)