Upload folder using huggingface_hub

app.py

@@ -1,14 +1,15 @@
 import gradio as gr
 import torch
-from diffusers import BriaFiboEditPipeline
+from diffusers import DiffusionPipeline
 from PIL import Image
 import os
 
 # Load the model
 print("Loading RMBG-3.0 model...")
-pipe = BriaFiboEditPipeline.from_pretrained(
+pipe = DiffusionPipeline.from_pretrained(
     "briaai/RMBG-3.0",
     torch_dtype=torch.bfloat16,
+    trust_remote_code=True
 )
 
 # Check if CUDA is available

requirements.txt

@@ -1,5 +1,5 @@
 torch>=2.0.0
-diffusers>=0.30.0
+git+https://github.com/huggingface/diffusers.git
 transformers>=4.40.0
 accelerate>=0.30.0
 safetensors>=0.4.0