Create dockerfile

dockerfile

@@ -0,0 +1,23 @@
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git ffmpeg libsm6 libxext6 wget curl && \
+    pip install --upgrade pip
+
+# Install Python dependencies
+COPY requirements.txt /app/requirements.txt
+WORKDIR /app
+RUN pip install -r requirements.txt
+
+# Copy source code and scripts
+COPY . /app
+
+# Make script executable
+RUN chmod +x /app/start.sh
+
+# Download huggingface_hub for model downloading
+RUN pip install huggingface_hub
+
+# Set entry point to your script
+ENTRYPOINT ["bash", "/app/start.sh"]