How to get unique classes from torch script model?

[maxrivera (Max)]

Is it possible to get the model.classes from a loaded TorchScript model? I want to use the map the probability index from the prediction to the class name.