Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -6,7 +6,8 @@ import numpy as np
 SER_SKETCH=Image.fromarray(np.zeros((1024,1024)))
 
 def get_img_from_sketchpad(img):
-    return Image.fromarray(img["composite"])
+    global SER_SKETCH
+    SER_SKETCH=Image.fromarray(img["composite"])
 
 def set_ser_sketch(img):
     global SER_SKETCH
@@ -47,7 +48,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                         inputs=[sketch, first_prompt, second_prompt],
                         label='Examples. Be concrete as much as possible.')
                 
-        sketch.input(get_img_from_sketchpad, outputs=set_ser_sketch, inputs=sketch, show_progress="hidden")
+        sketch.input(fn=get_img_from_sketchpad, inputs=sketch, show_progress="hidden")
         
         improve_sketch.click(fn=controller.get_first_result,
                              inputs=[SER_SKETCH, first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],