Update app.py

app.py

@@ -5,22 +5,16 @@ import io
 
 def remove_bg(image):
     img_bytes = io.BytesIO()
-    image.save(img_bytes, format="PNG")
-    output = remove(img_bytes.getvalue())
-    return Image.open(io.BytesIO(output))
+    image.save(img_bytes, format='PNG')
+    result = remove(img_bytes.getvalue())
+    return Image.open(io.BytesIO(result))
 
-with gr.Blocks(title="AI Background Remover") as demo:
-    gr.Markdown("### 🖼️ AI Background Remover\nUpload an image and remove background instantly using **Rembg**")
-
-    with gr.Row():
-        input_img = gr.Image(type="pil", label="Upload Image")
-        output_img = gr.Image(label="Result")
-
-    with gr.Row():
-        clear_btn = gr.ClearButton(input_img)
-        process_btn = gr.Button("Remove Background")
-
-    process_btn.click(fn=remove_bg, inputs=input_img, outputs=output_img)
+demo = gr.Interface(
+    fn=remove_bg,
+    inputs=gr.Image(type="pil", label="🖼️ Upload an Image"),
+    outputs=gr.Image(label="✨ Background Removed"),
+    title="🎨 AI Background Remover",
+    description="Upload an image and instantly remove the background using AI",
+)
 
 demo.launch()
-print("App running - built by Karthik")