Dockerfile

# Use official Python image
FROM python:3.9

# Set working directory
WORKDIR /code

# Copy requirements and install dependencies
# Note: We use the CPU version of PyTorch to keep the image small
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade pip
RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
RUN pip install --no-cache-dir -r /code/requirements.txt

# Copy the rest of the files
COPY . .

# Grant permissions (sometimes needed for HF)
RUN chmod -R 777 /code

# Run the app
CMD ["python", "app.py"]