Use if TYPE_CHECKING

Change-Id: I63c4c7909f7d7060a4b557f41cff4ba9010e004c

google/generativeai/vision_models/_vision_models.py

@@ -34,16 +34,22 @@
 from proto.marshal.collections import repeated
 
 
-# pylint: disable=g-import-not-at-top
-try:
+# pylint: disable=g-import-not-at-top\
+if typing.TYPE_CHECKING:
     from IPython import display as IPython_display
-except ImportError:
-    IPython_display = None
-
-try:
-    from PIL import Image as PIL_Image
-except ImportError:
-    PIL_Image = None
+else:
+    try:
+        from IPython import display as IPython_display
+    except ImportError:
+        IPython_display = None
+
+if typing.TYPE_CHECKING:
+    import PIL.Image as PIL_Image
+else:
+    try:
+        from PIL import Image as PIL_Image
+    except ImportError:
+        PIL_Image = None
 
 
 # This is to get around https://github.com/googleapis/proto-plus-python/issues/488