Update Dockerfile

Dockerfile

@@ -28,6 +28,8 @@ RUN git clone https://github.com/fffiloni/DemoFusion.git $HOME/app
 # Install dependencies
 RUN pip install --no-cache-dir accelerate gradio transformers==4.35.2  diffusers==0.21.4
 
+COPY app.py .
+
 RUN find $HOME/app
 
 # Set the environment variable to specify the GPU device
@@ -35,4 +37,4 @@ ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
 # Run your app.py script
-CMD ["python", "gradio_demo.py"]
+CMD ["python", "app.py"]