Improvement Suggestion

[darnells68]

I have two suggestions but first amazing bot man honestly simple and well made.

Okay the first suggestion I'd like to make is when the user clicks retry to generate more images with the same prompt. There is an issue where the new images are generated and it gives u the option to upscale the new images but not an option to retry again and again.

The second feature is it possible to generate 4 images at a time instead of the 2 with the necessary button for upscaling for the addition other two.

[IntenseSunlight]

I had a similar issue, where the "re-roll" feature generates 2 new images with only the possibility of upscaling. Also, there could be a possibility of 2 or 4 images (more probably doesn't make sense). The only issue is generation time would be longer. I can try to incorporate these items into my repo changes and make a PR here.