How to force model to generate image?

[haochuan-li]

Hi! Great work.

I see there's a "force image generation" option in the gradio demo. I wonder how to implement this in code? Can anyone enlighten me on this?

Thanks.