diff --git a/Dockerfile b/Dockerfile
index f21c056..c4810c0 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,16 +1,15 @@
-FROM nvcr.io/nvidia/l4t-tensorflow:r32.6.1-tf2.5-py3
+FROM  tensorflow/tensorflow:2.5.1
+
 RUN apt-get install make g++ gcc
-RUN pip3 install gunicorn 
+RUN pip3 install gunicorn
 RUN pip3 install setuptools==46.1.3
-#RUN pip3 install flask numpy Pillow six matplotlib
+
 RUN mkdir -p /app
 WORKDIR /app
 
 COPY requirements.txt /app
-RUN pip3 install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
+RUN pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
 
 COPY ./app /app
 EXPOSE 5000
 CMD ["gunicorn", "--bind", ":5000", "server:app"]
-
-
diff --git a/requirements.txt b/requirements.txt
index 5797356..80d59c1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,3 @@ Flask==1.1.2
 numpy==1.18.4
 Pillow==7.1.2
 six==1.15.0
-