Fix GPU task abort and duplicate processing

- Set @spaces.GPU(duration=300) to prevent ZeroGPU timeout (8B model needs ~3min)
- Add explicit pipe.to("cuda") inside GPU function
- Remove input_image.change handler causing duplicate GPU tasks
- Use BriaFiboEditPipeline directly (built into diffusers, trust_remote_code unnecessary)
- Move theme param to launch() per Gradio 6.0 API
- Add show_error=True for better error visibility

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

app.py

@@ -1,30 +1,27 @@
 import spaces
 import gradio as gr
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import BriaFiboEditPipeline
 from PIL import Image
 import tempfile
 
-# Load the model
+# Load the model on CPU at module level (ZeroGPU moves to CUDA inside @spaces.GPU)
 print("Loading Fibo-Edit-RMBG model...")
-pipe = DiffusionPipeline.from_pretrained(
+pipe = BriaFiboEditPipeline.from_pretrained(
     "briaai/Fibo-Edit-RMBG",
     torch_dtype=torch.bfloat16,
-    trust_remote_code=True,
 )
-
-device = "cuda" if torch.cuda.is_available() else "cpu"
-# pipe.to(device)
-print(f"Model loaded on {device}")
+print("Model loaded on CPU")
 
 INSTRUCTION = {'edit_instruction':'Generate a detailed grayscale alpha matte. Map the opaque foreground to white and the background to black. Produce soft, anti-aliased grayscale gradients at the edges of the subject to represent fine details and transparency.'}
 
 
-@spaces.GPU
+@spaces.GPU(duration=300)
 def process(image, num_steps=10, guidance_scale=1.0):
     if image is None:
         return None, None
 
+    pipe.to("cuda")
     mask = pipe(
         image=image,
         prompt=INSTRUCTION,
@@ -46,7 +43,7 @@ def process(image, num_steps=10, guidance_scale=1.0):
     return (image.convert("RGBA"), result), tmp.name
 
 
-with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal", theme=gr.themes.Soft(primary_hue="sky")) as demo:
+with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal") as demo:
     gr.Markdown("""
 # Fibo-Edit-RMBG - Background Removal
 
@@ -114,11 +111,5 @@ For commercial use, please [contact Bria AI](https://bria.ai/contact-us).
         outputs=[slider, download_file],
     )
 
-    input_image.change(
-        fn=process,
-        inputs=[input_image],
-        outputs=[slider, download_file],
-    )
-
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(show_error=True, theme=gr.themes.Soft(primary_hue="sky"))

requirements.txt

@@ -5,4 +5,4 @@ accelerate==1.13.0
 safetensors>=0.7.0
 pillow==12.1.1
 gradio==6.9.0
-spaces==0.47.0
+spaces==0.47.0