Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -38,7 +38,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                 
 
         improve_sketch.click(fn=controller.get_first_result,
-                             inputs=[Image.fromarray(sketch), first_prompt, first_negative_prompt],
+                             inputs=[Image.fromarray(sketch['composite']), first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],
                              outputs=improved_sketch_view)
 
         result_button.click(fn=controller.get_second_result,
@@ -46,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=[Image.fromarray(sketch)],
+                                 inputs=[Image.fromarray(sketch['composite'])], #[sketch],
                                  outputs=first_prompt)
 
         second_prompt_helper.click(fn=controller.get_help_w_prompt,