diff --git a/docker/Dockerfile b/docker/Dockerfile
index 3640ada..815f204 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -2,4 +2,7 @@ FROM python:3.9.17
 COPY . /digits/
 RUN pip3 install --no-cache-dir -r /digits/requirements.txt
 WORKDIR /digits
-CMD ["python3","digits.py","--runs=1","--test_size=0.2","--dev_size=0.3","--models=svm,tree"]
\ No newline at end of file
+EXPOSE 5000
+ENV FLASK_APP=api/app.py
+CMD ["run","flask","--host=0.0.0.0"]
+#CMD ["python3","digits.py","--runs=1","--test_size=0.2","--dev_size=0.3","--models=svm,tree"]
\ No newline at end of file