Dockerfile

FROM python:3.13-slim

# Set the working directory
WORKDIR /app

RUN apt-get update && apt-get install -y build-essential

# Update pip to the latest version
RUN pip install --upgrade pip

RUN pip install -U setuptools wheel

# Install torch
RUN pip install -U torch torchvision torchaudio

# Copy the requirements file into the container
COPY requirements.txt .

# Install the dependencies
RUN pip install --no-cache-dir -r requirements.txt

# Copy the rest of the application code into the container
COPY . .

EXPOSE 7860

ENTRYPOINT ["fastapi", "run", "main.py", "--host=0.0.0.0", "--port=7860"]