d96189d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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"]