I didn't say "assuming all criminals are black is a wise guess." What I meant to point out was that even if black people constitute even 51% of the prison population, the model would still be making a statistically-sound guess by returning an image of a black person.

Now if you asked for 100 images of criminals, and all of them were black, that would not be statistically-sound anymore.