Classification AI

Hi :slight_smile:
Is there any extension that I can use as classification AI system?
I’ve used ML4K before, but it has 100 pieces limitation… :frowning:
Any ways to use classification AI?
It’s not about MS AI extension, I want to use my own classification model.
I don’t want AI to label image as apple or pineapple.
I want AI to group images into 4~5 classes. You might know what I want via ML4K.
Here’s the example image:

Thanks!!!