I am using

CodePudding user response：

The model only outputs as numbers, each of the unique number representing a category.

In the above example, you need to create a map that assigns a Category to display name: {'0': 'Cat', '1': 'dog'}.