diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index 2198c81..3d3b26a 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -2,13 +2,12 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
 WORKDIR /subgen
 
-RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg curl
+RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg
 
 ENV PYTHONUNBUFFERED 1
 
-ADD https://raw.githubusercontent.com/McCloudS/subgen/main/bootstrap.sh /bootstrap.sh
-RUN chmod +x /bootstrap.sh
+ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py
 
-CMD [ "/bootstrap.sh" ]
+CMD [ "python3", "-u", "./subgen.py" ]
 
 EXPOSE 8090