How to cluster/predict images?

[laxmimerit]

Hi,
I have trained using examples given with pytorch-lightning. I couldn't find code to do clustering of images after training. How can I find which image falls in which cluster? Is there any predictor API?
I want to do something like this

[BakingBrains]

@laxmimerit anything you found out?

Thank you