diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index 3d3b26a..bcfeb1b 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -6,8 +6,9 @@ RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg
 
 ENV PYTHONUNBUFFERED 1
 
-ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py
+ADD https://raw.githubusercontent.com/McCloudS/subgen/main/bootstrap.sh /bootstrap.sh
+RUN chmod +x /bootstrap.sh
 
-CMD [ "python3", "-u", "./subgen.py" ]
+CMD [ "/bootstrap.sh" ]
 
 EXPOSE 8090