v2.2

app.py

@@ -12,11 +12,19 @@ login(os.getenv("HF_TOKEN"))
 @spaces.GPU()
 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")
+        print("\nDid not Receive Mask\n")
+        print("Mask Size : ", image["layers"][0].shape)
+        print("Unique values: ", np.unique(image["layers"][0]))
+        print("Type : ", type(image["layers"][0]))
         output = process_inpaint(image["background"], image["layers"][0])
     else: 
-        print("Processing Received Mask")
-        print("Mask : ", mask["background"][:,:,3])
+        print("\nProcessing Received Mask\n")
+        print("Mask Size : ", mask.shape)
+        print("Unique values: ", np.unique(mask))
+        print("Type : ", type(mask))
+        print("Mask Size : ", mask["background"].shape)
+        print("Unique values: ", np.unique(mask["background"]))
+        print("Type : ", type(mask["background"]))
         output = process_inpaint(image["background"], mask["background"])
     img_output = Image.fromarray(output).convert("RGB")