diff --git a/Dockerfile b/Dockerfile
index db5b7d03..d00dd75c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -422,6 +422,8 @@ RUN pip install flashtext && \
     pip install easyocr && \
     # Keep JAX version in sync with GPU image.
     pip install jax==0.2.12 jaxlib==0.1.64 && \
+    # ipympl adds interactive widget support for matplotlib
+    pip install ipympl==0.7.0 && \
     /tmp/clean-layer.sh
 
 # Download base easyocr models.