The output does not have category information, how can I print the category information on the image?