ajeetraina/flaskspace

1from transformers import pipeline 2from flask import Flask, jsonify, request, os3 4app = Flask(__name__)5 6nlp = pipeline("fill-mask", model=os.environ['MODEL_NAME'])7 8@app.route('/predict', methods=['POST'])9def predict():10    text = request.json['text']11    result = nlp(text)12    return jsonify(result[0]['sequence'])13 14if __name__ == '__main__':15    app.run(port=os.environ['API_PORT'])16 17