from transformers import pipeline
from flask import Flask, request, jsonify



app =Flask(__name__)
generator=pipeline('text-generation',model='gbt2')
@app.route('/generate',methods=['POST'])
def generate():
    data=request.json
    prompt = data.get("prompt", "")
    results = generator(prompt, max_length=50, num_return_sequences=1)
    return jsonify(results)

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)