Dockerfile

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"]