🌘w🌖

app.py

@@ -2,19 +2,11 @@ from transformers import pipeline
 import gradio as gr
 import spaces
 
-pipe = pipeline(
-    "image-segmentation",
-    model="briaai/RMBG-1.4",
-    revision="refs/pr/9",
-    trust_remote_code=True,
-)
-
+pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)
 
 @spaces.GPU
 def predict(image_path):
-    pipe(image_path, out_name="myout.png")
-    return "myout.png"
-
+    return pipe(image_path)
 
 gr.Interface(
     predict,