diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9e32f8ac2eb98cd96a34c25a681ee0c8691e9ace..d9073449f68abacefb7b3aca2860fcab1dcb537f 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -42,16 +42,12 @@ after_script:
 build:
   stage: build
   image: docker:latest
-#  image: docker:dind
   services:
     - name: docker:dind
-      command: ["--host=fd://", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
+      command: ["--host=tcp://0.0.0.0:2375", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
   script:
     - docker buildx create --use
-    #    - docker build --cache-from=flask-backend-full --no-cache --progress=plain -t flask-backend-full -f flask_ml_api/Dockerfile .
     - docker build --cache-from=22baar1mst/asyst-nlp-grader --no-cache --progress=plain -t 22baar1mst/asyst-nlp-grader:latest -f flask_ml_api/Dockerfile .
-#    - docker build --no-cache --progress=plain -t 22baar1mst/asyst-nlp-grader:latest -f flask_ml_api/Dockerfile .
-    - echo $DOCKER_REPO
     - docker push 22baar1mst/asyst-nlp-grader:latest
   only:
     - master
@@ -59,7 +55,6 @@ build:
 deploy:
   stage: deploy
   image: docker:latest
-#  image: docker:dind
   services:
     - name: docker:dind
       command: ["--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index 8944451858c40cb1c009eb8137bf67a7e4b4c7c1..bdc1fc03587f7ede942c101878f48d7fc00adcc8 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -1,4 +1,3 @@
-#FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime as build
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime as build
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -14,19 +13,11 @@ COPY . /app
 
 RUN pip install --no-cache-dir Flask matplotlib scikit-learn transformers pandas sentence_transformers && \
     pip install --no-cache-dir --upgrade setuptools wheel && \
-    rm -rf /root/.cache/pip \
-
-#RUN pip install --no-cache-dir Flask && \
-#    pip install --no-cache-dir matplotlib && \
-#    pip install --no-cache-dir scikit-learn && \
-#    pip install --no-cache-dir transformers && \
-#    pip install --no-cache-dir pandas && \
-#    pip install --no-cache-dir sentence_transformers && \
-#    pip install --no-cache-dir --upgrade setuptools wheel
-#RUN rm -rf /root/.cache/pip
+    rm -rf /root/.cache/pip
 
 
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
+
 COPY --from=build /opt/myenv /opt/myenv
 
 ENV PATH="/opt/myenv/bin:$PATH"