Update main.py

main.py

@@ -2,18 +2,18 @@ import gradio as gr
 from rembg import remove
 from PIL import Image
 
-
 def remove_background(input_image):
+    # Convert to PIL Image if not already
+    if not isinstance(input_image, Image.Image):
+        input_image = Image.fromarray(input_image)
+    
     output_image = remove(input_image)
     return output_image
 
-
-inputs = gr.inputs.Image()
-outputs = gr.outputs.Image(type="pil")
 interface = gr.Interface(
     fn=remove_background,
-    inputs=inputs,
-    outputs=outputs,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
     title="Remove Background",
     description="This App removes the background from an image",
     examples=[
@@ -23,4 +23,6 @@ interface = gr.Interface(
     ],
     cache_examples=True,
 )
-interface.launch(enable_queue=True)
+
+if __name__ == "__main__":
+    interface.launch(enable_queue=True)