ALIH213 / Dockerfile
dx
Install torchvision CPU-only in Dockerfile to match CPU torch and prevent nms operator failure
5d52171
Raw
History Blame Contribute Delete
867 Bytes
# Use official Python image
FROM python:3.10
# Install curl and other necessary tools
RUN apt-get update && apt-get install -y curl libsndfile1 && rm -rf /var/lib/apt/lists/*
# Set up user and directory (HF Spaces runs with UID 1000)
RUN useradd -m -u 1000 user
USER user
ENV PATH="/home/user/.local/bin:$PATH"
WORKDIR /app
# Install CPU PyTorch first (to save space and download size)
RUN pip install --no-cache-dir --upgrade pip && \
pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
# Install other python dependencies
COPY --chown=user ./requirements.txt requirements.txt
RUN pip install --no-cache-dir --upgrade -r requirements.txt
# Copy the rest of the application files
COPY --chown=user . /app
# Ensure startup script is executable
RUN chmod +x /app/start.sh
# Start the services
CMD ["/app/start.sh"]