show output image after removing the background

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from rembg import remove
 from PIL import Image
 import logging
-import os
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -23,19 +22,22 @@ def remove_background(input_image):
         output_image.save(output_path)
         logging.info(f"Saved output image {output_path}")
 
-        return output_path
+        return output_image, output_path
 
     except Exception as e:
         logging.error(f"An error occurred: {e}")
-        return None
+        return None, None
 
 # Gradio interface
 iface = gr.Interface(
     fn=remove_background,
     inputs=gr.Image(type="pil"),
-    outputs=gr.File(label="Download the output image"),
+    outputs=[
+        gr.Image(type="pil", label="Output Image"), 
+        gr.File(label="Download the output image")
+    ],
     title="Background Remover",
-    description="Upload an image to remove the background and download the result.",
+    description="Upload an image to remove the background. You can view the result and download it.",
     allow_flagging="never",
 )