Dockerfile

FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime

WORKDIR /app

RUN apt-get update && apt-get install -y git python3 python3-pip

COPY . /app

RUN pip install -r requirements.txt

EXPOSE 7860

CMD ["python3", "main.py"]