Upload 3 files

README.md

@@ -1,14 +1,11 @@
 ---
 title: Image Background Remover
 emoji: 📉
-colorFrom: yellow
-colorTo: purple
+colorFrom: gray
+colorTo: red
 sdk: gradio
-sdk_version: 5.5.0
+sdk_version: 4.36.1
 app_file: app.py
 pinned: false
-license: apache-2.0
-short_description: Give you Best result
+license: mit
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,29 @@
+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()
+    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(
+    fn=remove_background,
+    inputs=gr.Image(label="Upload Image"),
+    outputs=gr.Image(label="Output Image"),
+    title="Image Background Remover",
+    description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
+    theme="Yntec/HaleyCH_Theme_Orange"
+)
+
+if __name__ == "__main__":
+    iface.launch()

requirements.txt

@@ -0,0 +1 @@
+transparent-background