Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -7,7 +7,7 @@ import numpy as np
 
 def get_img_from_sketchpad(img):
     #global SER_SKETCH
-    return Image.fromarray(img["composite"])
+    return Image.fromarray((img["composite"]))#,'L')
 
 #def set_ser_sketch(img):
     #global SER_SKETCH
@@ -17,7 +17,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='numpy', image_mode="RGB", brush=gr.Brush()) #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')
                 ser_image=gr.Image(type="pil")#, visible=False)
                 
             with gr.Column():