Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -5,7 +5,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
     with gr.Blocks() as ui:
         with gr.Row():
             with gr.Column():
-                sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type = 'pil', image_mode='RGB') #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
+                sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type='numpy', image_mode='L', brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
                 
             with gr.Column():
                 improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")