def get_caption(self, image_name):
image_model = ImageModel(image_name)
caption = image_model.get_pred()
return caption
def get_caption(self, path):
image_model = ImageModel(path)
caption = image_model.get_pred()
return caption