Update app.py

app.py

@@ -1,11 +1,19 @@
 import spaces
 import gradio as gr
 from transparent_background import Remover
+from PIL import Image
+import numpy as np
 
 @spaces.GPU
 def remove_background(image):
     remover = Remover()
-    output = remover.process(image)
+    if isinstance(image, Image.Image):
+        output = remover.process(image)
+    elif isinstance(image, np.ndarray):
+        image_pil = Image.fromarray(image)
+        output = remover.process(image_pil)
+    else:
+        raise TypeError("Unsupported image type")
     return output
 
 iface = gr.Interface(
@@ -17,4 +25,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()