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')