Upload folder using huggingface_hub

app.py

@@ -17,16 +17,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe.to(device)
 print(f"Model loaded on {device}")
 
-def process_image(image, mask, instruction, num_steps, guidance_scale):
+def process_image(image, mask, instruction, num_steps=50, guidance_scale=5.0):
     """
     Process an image with RMBG-3.0
-
-    Args:
-        image: Input image (PIL Image)
-        mask: Optional mask image (PIL Image)
-        instruction: Text instruction for editing
-        num_steps: Number of inference steps
-        guidance_scale: Guidance scale for generation
     """
     if image is None:
         return None, "Please upload an image"

requirements.txt

@@ -5,4 +5,4 @@ accelerate>=0.30.0
 safetensors>=0.4.0
 pillow>=10.0.0
 huggingface_hub>=0.34.0
-gradio>=5.0.0
+gradio==5.7.1