From 5d863c01e29acdd951767724a5e759613c1ab783 Mon Sep 17 00:00:00 2001
From: Pavel Yakubovskiy <qubvel@gmail.com>
Date: Sat, 28 Sep 2019 22:24:26 +0300
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile b/Dockerfile
index 0d55e933..f3a415a3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -492,6 +492,7 @@ RUN pip install flashtext && \
     pip install pytorch-ignite && \
     pip install qgrid && \
     pip install bqplot && \
+    pip install segmentation-models-pytorch && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages