diff --git a/.github/workflows/on_push_deploy_to_staging.yaml b/.github/workflows/on_push_deploy_to_staging.yaml
index c879664b6..00397258f 100644
--- a/.github/workflows/on_push_deploy_to_staging.yaml
+++ b/.github/workflows/on_push_deploy_to_staging.yaml
@@ -6,7 +6,7 @@ on:
       - master
     paths:
       - ".github/workflows/on_push_deploy_to_staging.yaml"
-      # TODO: Decide if we want to add other file changes here. 
+      # TODO: Decide if we want to add other file changes here.
       # TODO: Decide how we handle updating this forked copy.
 jobs:
   deploy:
@@ -77,6 +77,8 @@ jobs:
             
             # Run the new Docker container
             docker run --gpus all -d -p 7861:7861 --name moonlite-sd \
+              -v /home/ec2-user/apps/stable-diffusion-webui/outputs/img2img-images:/outputs/img2img-images \
+              -v /home/ec2-user/apps/media_root:/home/ec2-user/apps/media_root \
               --env-file /home/${{ secrets.STAGING_EC2_USER }}/.env \
               ${{ secrets.STAGING_ECR_URI }}:latest
             
diff --git a/.github/workflows/on_tag_deploy_to_production.yaml b/.github/workflows/on_tag_deploy_to_production.yaml
index 2c724834e..1fd4cd20e 100644
--- a/.github/workflows/on_tag_deploy_to_production.yaml
+++ b/.github/workflows/on_tag_deploy_to_production.yaml
@@ -74,6 +74,8 @@ jobs:
             
             # Run the new Docker container
             docker run --gpus all -d -p 7861:7861 --name moonlite-sd \
+              -v /home/ec2-user/apps/stable-diffusion-webui/outputs/img2img-images:/outputs/img2img-images \
+              -v /home/ec2-user/apps/media_root:/home/ec2-user/apps/media_root \
               --env-file /home/${{ secrets.PROD_EC2_USER }}/.env \
               ${{ secrets.PROD_ECR_URI }}:latest
             
diff --git a/Dockerfile b/Dockerfile
index 29fd3f838..c3a0fe948 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,6 +8,10 @@ ENV WORKDIR /app
 RUN mkdir -p $WORKDIR
 WORKDIR $WORKDIR
 
+# Create the outputs/img2img-images folder
+RUN mkdir -p /outputs/img2img-images
+RUN mkdir -p /home/ec2-user/apps/media_root
+
 RUN apt-get update
 
 # Install Python, pip, and other necessary dependencies