Awesome project. Exactly what I was looking for.

When it produces a set of images for a given prompt, wouldnt it be better if we could remove a set of images from the possible selection ? Does it not work this way? Another idea would be to provide a few different kinds of prompts and based on that select all the images that matter for the given "class".

Some other things that would be good to know:

1. Can we keep adding items to the classifier? and getting newer versions of the classifier with the newly added item ? 2. How to deploy and host this kind of models? Is there any guidelines on how to deploy this in AWS or GCS for production use cases ?