Upload folder using huggingface_hub

app.py

@@ -30,15 +30,24 @@ def process_image(image, instruction, num_steps=50, guidance_scale=5.0):
         return None, "Please provide an editing instruction"
 
     try:
-        # Run the pipeline
-        result = pipe(
+        # Run the pipeline to get the grayscale alpha matte
+        mask = pipe(
             image=image,
             prompt=instruction,
             num_inference_steps=num_steps,
             guidance_scale=guidance_scale
         ).images[0]
 
-        return result, "✅ Image processed successfully!"
+        # Convert original image to RGBA
+        original_rgba = image.convert("RGBA")
+
+        # Convert mask to grayscale (in case it's not already)
+        mask_gray = mask.convert("L")
+
+        # Use the mask as the alpha channel
+        original_rgba.putalpha(mask_gray)
+
+        return original_rgba, "✅ Image processed successfully!"
 
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
@@ -75,9 +84,9 @@ with gr.Blocks(title="RMBG-3.0 - Background Removal & Image Editing") as demo:
 
             with gr.Accordion("Advanced Settings", open=False):
                 num_steps = gr.Slider(
-                    minimum=4,
-                    maximum=100,
-                    value=50,
+                    minimum=1,
+                    maximum=50,
+                    value=4,
                     step=1,
                     label="Number of Steps"
                 )