Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official PyTorch image with CUDA support as the base image
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
 
 # Install Git and system libraries required for OpenGL without interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
@@ -14,15 +14,15 @@ RUN useradd -m -u 1000 user
 USER user
 
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+    PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=$HOME/app \
-	PYTHONUNBUFFERED=1 \
-	GRADIO_ALLOW_FLAGGING=never \
-	GRADIO_NUM_PORTS=1 \
-	GRADIO_SERVER_NAME=0.0.0.0 \
-	GRADIO_THEME=huggingface \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
     GRADIO_SHARE=False \
-	SYSTEM=spaces
+    SYSTEM=spaces
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
@@ -30,8 +30,11 @@ WORKDIR $HOME/app
 # Clone your repository or add your code to the container
 RUN git clone -b main https://github.com/fffiloni/Hotshot-XL $HOME/app
 
+# Add the wrapper script
+COPY run_app.py $HOME/app/
+
 # Install dependencies
-RUN pip install --no-cache-dir -r requirements.txt gradio accelerate
+RUN pip install --no-cache-dir -r requirements.txt gradio accelerate gradio-auth
 
 RUN find $HOME/app
 
@@ -39,5 +42,5 @@ RUN find $HOME/app
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
-# Run your app.py script
-CMD ["python", "app_w_lora.py"]
+# Run the wrapper script
+CMD ["python", "run_app.py"]