From 23045b1663d57610940fbcd11378bd0ea19e60da Mon Sep 17 00:00:00 2001 From: BenAAndrew Date: Fri, 8 Oct 2021 13:33:30 +0100 Subject: [PATCH] Fix docker install --- Dockerfile | 20 ++++++++++++-------- main.py | 2 +- 2 files changed, 13 insertions(+), 9 deletions(-) diff --git a/Dockerfile b/Dockerfile index 66e2d04..2ec0991 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,16 +1,20 @@ -FROM pytorch/pytorch:1.7.0-cuda11.0-cudnn8-runtime +FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime + +# Lib dependencies +RUN apt-get update +RUN apt-get install -y ffmpeg build-essential # Setup -RUN apt-get update WORKDIR /app -COPY . /app - -# FFMPEG -RUN apt-get install -y ffmpeg +COPY application/ /app/application +COPY dataset/ /app/dataset +COPY training/ /app/training +COPY synthesis/ /app/synthesis +COPY main.py /app +COPY requirements.txt /app # Python dependencies RUN pip install -r requirements.txt # Start app -ENTRYPOINT [ "python3" ] -CMD [ "main.py" ] +CMD [ "python3", "main.py" ] diff --git a/main.py b/main.py index 3a6d238..755cd66 100644 --- a/main.py +++ b/main.py @@ -50,4 +50,4 @@ if __name__ == "__main__": cleanup_mei() check_ffmpeg() webbrowser.open_new_tab("http://localhost:5000") - socketio.run(app) + socketio.run(app, host='0.0.0.0')