Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -17,8 +17,8 @@ def ui(controller):#get_first_result,get_second_result): #controller):
     with gr.Blocks() as ui:
         with gr.Row():
             with gr.Column():
-                sketch=gr.Sketchpad(interactive=True, label='Sketch', type='numpy', image_mode="L", brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
-                ser_image=gr.Image(type="pil")#, visible=False)
+                sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type='numpy', image_mode="L", brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
+                ser_image=gr.Image(type="pil", progress=gr.Progress())#, visible=False)
                 
             with gr.Column():
                 improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")        
@@ -53,7 +53,8 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                         label='Examples. Be concrete as much as possible.')
                 
         sketch.change(fn=get_img_from_sketchpad, inputs=sketch, outputs=ser_image, show_progress="full")
-        
+
+              
         improve_sketch.click(fn=controller.get_first_result,
                              inputs=[ser_image, first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],
                              outputs=improved_sketch_view)