install ipympl

Adds widget support for matplotlib

http://b/185060120

Author: djherbis

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 && \
     /tmp/clean-layer.sh
 
 # Download base easyocr models.