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