Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -4,7 +4,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
     with gr.Blocks() as ui:
         with gr.Row():
             with gr.Column():
-                sketch=gr.Image(sources = 'upload', label='Sketch', type = 'pil', tool='sketch')#gr.Sketchpad(sources = 'upload', label='Sketch', type = 'pil') #
+                sketch=gr.ImageEditor(sources = 'upload', label='Sketch', type = 'pil') #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
                 
             with gr.Column():
                 improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")