Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 
 import gradio as gr
 from pnginfo import read_info_from_image, send_paras
+from PIL import Image
 
 from utils import set_token, generate_novelai_image, image_from_bytes
 
@@ -120,7 +121,7 @@ def main_ui():
     gen_btn.click(generate, paras + [others[0], others[4]], [image, info])
     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[5].click(lambda i: gr.ImageEditor(value=i), inputs=image, outputs=paras[17])
+    others[5].click(lambda i: gr.ImageEditor(value=Image.new('RGBA', (256,256), 'red')), inputs=image, outputs=paras[17])
     return page, paras[:14]