mirror of
https://github.com/jantic/DeOldify.git
synced 2026-08-29 02:10:55 +08:00
jqueguiner
33 lines
444 B
Plaintext
33 lines
444 B
Plaintext
From nvcr.io/nvidia/pytorch:19.04-py3
|
|
|
|
RUN apt-get -y update
|
|
|
|
RUN apt-get install -y python3-pip software-properties-common wget ffmpeg
|
|
|
|
RUN apt-get -y update
|
|
|
|
RUN mkdir -p /root/.torch/models
|
|
|
|
RUN mkdir -p /data/models
|
|
|
|
ADD . /data/
|
|
|
|
WORKDIR /data
|
|
|
|
RUN pip install -r requirements.txt
|
|
|
|
RUN pip install Flask
|
|
|
|
RUN pip install Pillow
|
|
|
|
RUN pip install scikit-image
|
|
|
|
RUN pip install requests
|
|
|
|
EXPOSE 5000
|
|
|
|
ENTRYPOINT ["python3"]
|
|
|
|
CMD ["app.py"]
|
|
|