Added

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.8
+
+RUN pip install flask transformers
+
+COPY . .
+
+ENV MODEL_NAME bert-base-cased
+ENV API_PORT 8080
+
+CMD ["python", "app.py"]
+

app.py

@@ -0,0 +1,16 @@
+from transformers import pipeline
+from flask import Flask, jsonify, request
+
+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'])
+

requirements.txt

@@ -0,0 +1 @@
+flask