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