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"]