Update app.py

app.py

@@ -6,9 +6,9 @@ from PIL import Image
 import os
 
 # Load the model
-print("Loading RMBG-3.0 model...")
+print("Loading Fibo-Edit-RMBG model...")
 pipe = DiffusionPipeline.from_pretrained(
-    "briaai/RMBG-3.0",
+    "briaai/Fibo-Edit-RMBG",
     torch_dtype=torch.bfloat16,
     trust_remote_code=True
 )
@@ -19,9 +19,9 @@ pipe.to(device)
 print(f"Model loaded on {device}")
 
 @spaces.GPU
-def process_image(image, num_steps=50, guidance_scale=5.0):
+def process_image(image, num_steps=10, guidance_scale=1.0):
     """
-    Process an image with RMBG-3.0
+    Process an image with Fibo-Edit-RMBG
     """
     if image is None:
         return None, "Please upload an image"
@@ -57,11 +57,11 @@ def process_image(image, num_steps=50, guidance_scale=5.0):
         return None, f"❌ Error: {str(e)}"
 
 # Create Gradio interface
-with gr.Blocks(title="RMBG-3.0 - Background Removal") as demo:
+with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal") as demo:
     gr.Markdown("""
-    # RMBG-3.0 - Background Removal
+    # Fibo-Edit-RMBG - Background Removal
 
-    Powered by [Bria AI's RMBG-3.0 model](https://huggingface.co/briaai/RMBG-3.0)
+    Powered by [Bria AI's Fibo-Edit-RMBG model](https://huggingface.co/briaai/Fibo-Edit-RMBG)
 
     This model performs background removal on any image
 
@@ -83,7 +83,7 @@ with gr.Blocks(title="RMBG-3.0 - Background Removal") as demo:
                 num_steps = gr.Slider(
                     minimum=1,
                     maximum=50,
-                    value=4,
+                    value=10,
                     step=1,
                     label="Number of Steps"
                 )
@@ -91,7 +91,7 @@ with gr.Blocks(title="RMBG-3.0 - Background Removal") as demo:
                 guidance_scale = gr.Slider(
                     minimum=1.0,
                     maximum=15.0,
-                    value=5.0,
+                    value=1.0,
                     step=0.5,
                     label="Guidance Scale"
                 )
@@ -121,12 +121,12 @@ with gr.Blocks(title="RMBG-3.0 - Background Removal") as demo:
 
     gr.Markdown("""
     ---
-    ### About RMBG-3.0
+    ### About Fibo-Edit-RMBG
 
-    RMBG-3.0 is built on [Fibo-Edit](https://huggingface.co/briaai/Fibo-Edit), an 8B parameter image editing model
+    Fibo-Edit-RMBG is built on [Fibo-Edit](https://huggingface.co/briaai/Fibo-Edit), an 8B parameter image editing model
     that uses structured prompts for precise, deterministic editing.
 
-    - **Model**: [briaai/RMBG-3.0](https://huggingface.co/briaai/RMBG-3.0)
+    - **Model**: [briaai/Fibo-Edit-RMBG](https://huggingface.co/briaai/Fibo-Edit-RMBG)
     - **Architecture**: Based on FIBO with 8B parameters
     - **License**: Non-commercial use ([CC BY-NC 4.0](https://creativecommons.org/licenses/by-nc/4.0/))