🌘w🌖

README.md

@@ -7,6 +7,7 @@ sdk: gradio
 sdk_version: 4.18.0
 app_file: app.py
 pinned: false
+short_description: background removal using python and the transformers library
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,5 +1,6 @@
 from transformers import pipeline
 import gradio as gr
+import spaces
 
 pipe = pipeline(
     "image-segmentation",
@@ -9,6 +10,7 @@ pipe = pipeline(
 )
 
 
+@spaces.GPU
 def predict(image_path):
     pipe(image_path, out_name="myout.png")
     return "myout.png"
@@ -18,6 +20,6 @@ gr.Interface(
     predict,
     gr.Image(type="filepath"),
     "image",
-    examples=[["./lain with background.jpg"], ["./gigachad.webp"]],
+    examples=[["./hafedh purple.jpg"], ["./gigachad.webp"]],
     cache_examples=True,
 ).launch()

requirements.txt

@@ -1,5 +1,6 @@
-torch
+torch==2.2.0
 torchvision
 transformers
 pillow
-scikit-image
+scikit-image
+spaces