Update app.py

app.py

@@ -6,15 +6,7 @@ import numpy as np
 
 def remove_bg(image):
     try:
-        # Convert input to PIL Image
-        if isinstance(image, str):  # file path
-            img = Image.open(image)
-        elif isinstance(image, np.ndarray):  # numpy array from Gradio
-            img = Image.fromarray(image)
-        else:
-            raise ValueError("Unsupported input type")
-        
-        # Process with rembg
+        img = Image.open(image) if isinstance(image, str) else Image.fromarray(image)
         img_byte_arr = io.BytesIO()
         img.save(img_byte_arr, format='PNG')
         output_bytes = remove(img_byte_arr.getvalue())
@@ -22,16 +14,16 @@ def remove_bg(image):
     except Exception as e:
         raise gr.Error(f"Error processing image: {str(e)}")
 
-# Create interface with correct flagging mode
-iface = gr.Interface(
-    fn=remove_bg,
-    inputs=gr.Image(type="filepath", label="Input Image"),
-    outputs=gr.Image(label="Output Image", type="pil"),
-    title="Background Remover",
-    description="Upload an image to remove its background",
-    flagging_mode="never"  # Correct way to disable flagging
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# Background Remover")
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="filepath", label="Input Image")
+            submit_btn = gr.Button("Remove Background")
+        with gr.Column():
+            output_image = gr.Image(label="Output Image", type="pil")
+    
+    submit_btn.click(remove_bg, inputs=input_image, outputs=output_image)
 
-# Launch the app (API is automatically enabled in newer Gradio versions)
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)