flask deployment

Author: marcin-laskowski

Dockerfile

@@ -0,0 +1,21 @@
+# Use python as base image
+FROM python:3.6-stretch
+
+# Use working directory /app/model
+WORKDIR /app/model
+
+# Copy and install required packages
+COPY requirements.txt .
+RUN pip install --trusted-host pypi.python.org -r requirements.txt
+
+# Copy all the content of current directory to working directory
+COPY . .
+
+# Set env variables for Cloud Run
+ENV PORT 5000
+ENV HOST 0.0.0.0
+
+EXPOSE 5000:5000
+
+# Run flask app
+CMD ["python","app.py"]

app.py

@@ -0,0 +1,23 @@
+import json
+from flask import Flask, request
+from syndicai import PythonPredictor
+
+app = Flask(__name__)
+
+
+@app.route('/')
+def hello():
+    """ Main page of the app. """
+    return "Hello World!"
+
+
+@app.route('/predict')
+def predict():
+    """ Return JSON serializable output from the model """
+    payload = request.args
+    classifier = PythonPredictor("")
+    return classifier.predict(payload)
+
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000)

docker-compose.yaml

@@ -0,0 +1,7 @@
+version: '3'
+services:
+
+  flask_classifier:
+    build: .
+    ports:
+      - 5000:5000

requirements.txt

@@ -1,2 +1,4 @@
-torch
-torchvision
+torch==1.7.1
+torchvision==0.8.2
+flask==1.1.2
+requests==2.18.4