Any ai model that would take weather conditions as input and retrieve an image of the weather?

title. i’m trying to do it for small devices