Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -18,7 +18,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
         with gr.Row():
             with gr.Column():
                 sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type='pil', image_mode="RGB", brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
-                ser_image=gr.Image(type="pil")#, visible=False)
+                #ser_image=gr.Image(type="pil")#, visible=False)
                 
             with gr.Column():
                 improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")        
@@ -56,7 +56,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
 
               
         improve_sketch.click(fn=controller.get_first_result,
-                             inputs=[ser_image, first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],
+                             inputs=[sketch, first_prompt, first_negative_prompt], #[ser_image, first_prompt, first_negative_prompt],
                              outputs=improved_sketch_view)
 
         result_button.click(fn=controller.get_second_result,