Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
-#FROM mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun
+#FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
+FROM mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun
 #FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 WORKDIR /content
 ENV PATH="/home/zebraslive/.local/bin:${PATH}"
@@ -52,4 +52,4 @@ COPY --from=proxy readme.html /usr/share/nginx/html/readme.html
 # Copy the README.md
 COPY README.md /usr/share/nginx/html/README.md
 WORKDIR /content
-
+CMD ["python3", "/content/worker_runpod.py"]