run_app.py

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official PyTorch image with CUDA support as the base image
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 # Install Git and system libraries required for OpenGL without interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
@@ -30,11 +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/
+# Add the server script
+COPY server.py $HOME/app/
 
 # Install dependencies
-RUN pip install --no-cache-dir -r requirements.txt gradio accelerate gradio-auth
+RUN pip install --no-cache-dir -r requirements.txt gradio accelerate flask
 
 RUN find $HOME/app
 
@@ -42,5 +42,5 @@ RUN find $HOME/app
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
-# Run the wrapper script
+# Run the server script
 CMD ["python", "run_app.py"]