Update app.py

app.py

@@ -5,29 +5,33 @@ import io
 import numpy as np
 
 def remove_bg(image):
-    # 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_byte_arr = io.BytesIO()
-    img.save(img_byte_arr, format='PNG')
-    output_bytes = remove(img_byte_arr.getvalue())
-    return Image.open(io.BytesIO(output_bytes))
+    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_byte_arr = io.BytesIO()
+        img.save(img_byte_arr, format='PNG')
+        output_bytes = remove(img_byte_arr.getvalue())
+        return Image.open(io.BytesIO(output_bytes))
+    except Exception as e:
+        raise gr.Error(f"Error processing image: {str(e)}")
 
-# Define the interface
+# Define the interface with explicit API enabled
 iface = gr.Interface(
     fn=remove_bg,
-    inputs=gr.Image(type="filepath", label="Input Image"),  # Important: use "filepath" for API compatibility
-    outputs=gr.Image(label="Output Image"),
+    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"
+    description="Upload an image to remove its background",
+    allow_flagging="never"  # Disable the flagging feature
 )
 
-# Launch the app
+# Launch the app with API enabled
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    iface.launch(server_name="0.0.0.0", server_port=7860, enable_api=True)