Update app.py

app.py

@@ -88,7 +88,7 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
     today_count += 1
     img = image_from_bytes(img_data)
     
-    return img, payload
+    return gr.Image(value=img, height=height*721//width), payload
 
 
 def preview_ui():