Update app.py

app.py

@@ -189,6 +189,7 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
     images = np.array(images)
     masks = np.array(masks)
     img_tensor, mask_tensor = preprocess_for_removal(images, masks)
+    print("mask_tensor shape:",mask_tensor.shape)
     img_tensor=img_tensor.to("cuda")
     mask_tensor=mask_tensor.to("cuda")
     mask_tensor = mask_tensor[:,:,:]