Create Dockerfile

Dockerfile

@@ -0,0 +1,22 @@
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+
+# System utilities
+RUN apt-get update && apt-get install -y \
+    git ffmpeg libsm6 libxext6 curl wget
+
+# Set work directory
+WORKDIR /app
+
+# Copy code
+COPY . /app
+
+# Install Python packages
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
+RUN pip install huggingface_hub
+
+# Make the script executable
+RUN chmod +x /app/start.sh
+
+# Run startup script
+ENTRYPOINT ["bash", "/app/start.sh"]