Update app.py

app.py

@@ -108,7 +108,7 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
     today_count += 1
     img = image_from_bytes(img_data)
     filename = str(today_count).rjust(5, '0') + '-' + str(payload['parameters']['seed']) + '.png'
-    return gr.Image(value=img, label=filename), payload
+    return img, payload
 
 
 def preview_ui():
@@ -125,7 +125,7 @@ def main_ui():
                 gen_btn, paras, others = control_ui()
             with gr.Column():
                 image, info = preview_ui()
-    gen_btn.click(generate, paras + [others[0]], [image, info])
+    gen_btn.click(generate, paras + [others[0]], [image, info], postprocess=False)
     others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
     others[3].click(lambda i: i, inputs=image, outputs=paras[14])
     others[4].click(lambda i: i, inputs=image, outputs=paras[17])