Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -8,6 +8,10 @@ SER_SKETCH=Image.fromarray(np.zeros((1024,1024)))
 def get_img_from_sketchpad(img):
     return Image.fromarray(img["composite"])
 
+def set_ser_sketch(img):
+    global SER_SKETCH
+    SER_SKETCH=img
+
 def ui(controller):#get_first_result,get_second_result): #controller):
     with gr.Blocks() as ui:
         with gr.Row():
@@ -43,7 +47,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=SER_SKETCH, inputs=sketch, show_progress="hidden")
+        sketch.input(get_img_from_sketchpad, outputs=set_ser_sketch, 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],