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(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", visible=False)
+                sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type='numpy', image_mode="RGB", brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
+                ser_image=gr.Image(type="pil")#, visible=False)
                 
             with gr.Column():
                 improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")