def METHOD_NAME(self):
    aiplatform.init(project=e2e_base._PROJECT, location=e2e_base._LOCATION)
    model = ga_vision_models.ImageCaptioningModel.from_pretrained("imagetext")
    image = _create_blank_image()
    captions = model.get_captions(
        image=image,
        # Optional:
        number_of_results=2,
        language="en",
    )
    assert len(captions) == 2