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