v1.8

README.md

@@ -1,8 +1,8 @@
 ---
 title: Object Remover
 emoji: ⚡
-colorFrom: gray
-colorTo: gray
+colorFrom: indigo
+colorTo: green
 sdk: gradio
 sdk_version: 5.23.1
 app_file: app.py

app.py

@@ -10,9 +10,15 @@ import spaces
 login(os.getenv("HF_TOKEN"))
 
 @spaces.GPU()
-def process_image(image, progress=gr.Progress(track_tqdm=True)):
-    output = process_inpaint(image["background"], image["layers"][0])
+def process_image(image, mask, progress=gr.Progress(track_tqdm=True)):
+    if np.unique(mask["background"]).size == 1:
+        print("\n Did not Receive Mask\n")
+        output = process_inpaint(image["background"], image["layers"][0])
+    else: 
+        print("Processing Received Mask")
+        output = process_inpaint(image["background"], mask["background"])
     img_output = Image.fromarray(output).convert("RGB")
+    
     return img_output
 
 
@@ -21,9 +27,11 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             image = gr.ImageMask(type="numpy", layers=False, label="Upload Image")
+            with gr.Accordion(label="Advanced", open=False):
+                mask = gr.ImageMask(type="numpy", layers=False, label="Upload Mask")
             button = gr.Button("Remove")
         with gr.Column():
             output = gr.Image(format="png", label="Output Image")
-        button.click(fn=process_image, inputs=[image], outputs=[output])
+        button.click(fn=process_image, inputs=[image, mask], outputs=[output])
 
 demo.launch(debug=True)