changes?

app.py

@@ -41,7 +41,7 @@ with gr.Blocks(theme=theme) as GenDemo:
                    )
             with gr.Column():
                 button_gen = gr.Button("Generate Image", elem_id="generateIm", variant="primary")
-                gen_image = gr.Image(label="Generated Image", show_download_button=True, show_label=False)
+                gen_image = gr.Image(label="Generated Image", image_mode="RGBA", type='pil', show_download_button=True, show_label=False)
 
         button_gen.click(check_prompt, inputs=[prompt]).success(generate_image, inputs=[prompt, negative_prompt, control_image, scale], outputs=[gen_image])
 

util/text_img.py

@@ -44,7 +44,6 @@ def generate_image(prompt, negative_prompt, control_image, scale=0.5):
         controlnet_conditioning_scale=scale,
     ).images[0]
     image2 = rembg.remove(image)
-    print(image2)
 
     return image2