File size: 193 Bytes
df3934d
 
 
 
 
8d51f54
67b77da
1
2
3
4
5
6
7
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
WORKDIR /app
COPY requirements.txt .
RUN pip install -r requirements.txt
COPY . . 
EXPOSE 7860
CMD ["flask","run","--host=0.0.0.0", "-p 7860"]