diff --git a/Dockerfile b/Dockerfile
index 3e24888..ffc826d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,5 @@
-FROM continuumio/conda-ci-linux-64-python3.8
-USER root
+FROM tensorflow/tensorflow:2.2.0
 COPY . /app/
-RUN conda install -c anaconda tensorflow=2.2.0 pip
 RUN pip install -r /app/requirements.txt
 EXPOSE 5000
 WORKDIR /app
diff --git a/Dockerfile-conda b/Dockerfile-conda
new file mode 100644
index 0000000..3e24888
--- /dev/null
+++ b/Dockerfile-conda
@@ -0,0 +1,8 @@
+FROM continuumio/conda-ci-linux-64-python3.8
+USER root
+COPY . /app/
+RUN conda install -c anaconda tensorflow=2.2.0 pip
+RUN pip install -r /app/requirements.txt
+EXPOSE 5000
+WORKDIR /app
+ENTRYPOINT ["python3","app.py"]
\ No newline at end of file