Improve docstring of image classifier model spec.

PiperOrigin-RevId: 515466722
2023-03-09 15:59:45 -08:00 · 2023-03-09 15:59:45 -08:00 · 2d8f937913
commit 2d8f937913
parent 517e997179
1 changed files with 1 additions and 1 deletions
--- a/mediapipe/model_maker/python/vision/image_classifier/model_spec.py
+++ b/mediapipe/model_maker/python/vision/image_classifier/model_spec.py
@ -28,7 +28,7 @@ class ModelSpec(object):
               uri: str,
               input_image_shape: Optional[List[int]] = None,
               name: str = ''):
-    """Initializes a new instance of the `ImageModelSpec` class.
+    """Initializes a new instance of the image classifier `ModelSpec` class.

    Args:
      uri: str, URI to the pretrained model.