Upload folder using huggingface_hub

app.py

@@ -3,6 +3,7 @@ import torch
 from diffusers import DiffusionPipeline
 from PIL import Image
 import os
+import spaces
 
 # Load the model
 print("Loading RMBG-3.0 model...")
@@ -17,6 +18,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe.to(device)
 print(f"Model loaded on {device}")
 
+@spaces.GPU
 def process_image(image, instruction, num_steps=50, guidance_scale=5.0):
     """
     Process an image with RMBG-3.0

requirements.txt

@@ -6,3 +6,4 @@ safetensors>=0.4.0
 pillow>=10.0.0
 huggingface_hub>=0.34.0
 gradio>=5.8.0
+spaces