Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from PIL import Image
 
 def ui(controller):#get_first_result,get_second_result): #controller):
     with gr.Blocks() as ui:
@@ -37,7 +38,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                 
 
         improve_sketch.click(fn=controller.get_first_result,
-                             inputs=[sketch, first_prompt, first_negative_prompt],
+                             inputs=[Image.fromarray(sketch), first_prompt, first_negative_prompt],
                              outputs=improved_sketch_view)
 
         result_button.click(fn=controller.get_second_result,
@@ -45,7 +46,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                              outputs=result)
 
         firts_prompt_helper.click(fn=controller.get_help_w_prompt,
-                                 inputs=[sketch],
+                                 inputs=[Image.fromarray(sketch)],
                                  outputs=first_prompt)
 
         second_prompt_helper.click(fn=controller.get_help_w_prompt,