feat: ✨ add Flask Dockerfile

Dan6erbond · May 13, 2023 · 3c2198f · 3c2198f
1 parent ec0de56
commit 3c2198f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/tagger/Dockerfile b/tagger/Dockerfile
@@ -0,0 +1,10 @@
+FROM python:3.10-slim-buster
+
+WORKDIR /tagger
+
+COPY requirements.txt requirements.txt
+RUN pip3 install -r requirements.txt
+
+COPY . .
+
+CMD [ "python3", "-m" , "flask", "run", "--host=0.0.0.0"]
diff --git a/tagger/app.py b/tagger/app.py
@@ -75,3 +75,6 @@ def start_tag_file(file_id):
     thread.start()
 
     return {"file": file_id, "model": "facebook/detr-resnet-50"}, 202
+
+if __name__ == "__main__":
+    app.run()