diff --git a/Dockerfile b/Dockerfile
index 6b1b7c1..703b0a7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -52,7 +52,7 @@ RUN . /opt/conda/etc/profile.d/conda.sh && conda activate omni && \
 
 # Run download.py to fetch model weights and convert safetensors to .pt format
 RUN . /opt/conda/etc/profile.d/conda.sh && conda activate omni && \
-    python download2.py && \
+    python download.py && \
     echo "Contents of weights directory:" && \
     ls -lR weights && \
     python weights/convert_safetensor_to_pt.py
@@ -67,7 +67,7 @@ ENV GRADIO_SERVER_NAME="0.0.0.0"
 COPY entrypoint.sh /usr/src/app/entrypoint.sh
 RUN chmod +x /usr/src/app/entrypoint.sh
 
-# Keep the container running (WARNING: disables entrypoint)
+# To debug, keep the container running
 # CMD ["tail", "-f", "/dev/null"]
 
 # Set the entrypoint
diff --git a/deploy_requirements.txt b/deploy_requirements.txt
index da93605..82ad3b2 100644
--- a/deploy_requirements.txt
+++ b/deploy_requirements.txt
@@ -2,10 +2,11 @@ boto3==1.34.18
 botocore==1.34.18
 fire==0.5.0
 gitpython==3.1.41
-gradio_client==0.17.0
+gradio_client==1.4.2
 jinja2==3.1.3
 loguru==0.7.2
 paramiko==3.5.0
+Pillow==11.0.0
 pydantic_settings==2.1.0
 pynacl==1.5.0
 requests==2.31.0