v2.5

app.py

@@ -24,7 +24,7 @@ def process_image(image, mask, progress=gr.Progress(track_tqdm=True)):
         print("Unique values: ", np.unique(mask["background"]))
         print("Type : ", type(mask["background"]))
         mask = mask["background"]
-        output = process_inpaint(image["background"], mask["background"])
+        output = process_inpaint(image["background"], mask)
     img_output = Image.fromarray(output).convert("RGB")
     
     return img_output, mask