better mask matching

app.py

@@ -125,11 +125,11 @@ def inpaint(input_img, input_img_with_mask, prompt):
     input_img_256 = input_img.convert('RGB').resize((256, 256), resample=Image.BICUBIC)
     input_img_64 = input_img.convert('RGB').resize((64, 64), resample=Image.BICUBIC)
 
-    input_img_with_mask_converted = input_img.convert('RGBA')
-    input_img_with_mask_64 = input_img_with_mask_converted.getchannel( 'A' ) # Mode 'L'
+    input_img_with_mask_converted = input_img.convert('RGBA').input_img_with_mask_converted.getchannel( 'A' )  # Mode 'L'
 
-    input_img_with_mask_64 = input_img_with_mask_64.resize((64, 64), resample=Image.BICUBIC)
-    return input_img, input_img_with_mask_64
+    input_img_with_mask_64 = input_img_with_mask_converted.resize((64, 64), resample=Image.BICUBIC)
+    input_img_with_mask_256 = input_img_with_mask_converted.resize((256, 256), resample=Image.BICUBIC)
+    # return input_img, input_img_with_mask_64
     # Source image we are inpainting
     source_image_256 = pil_to_numpy(input_img_256)
     source_image_64 = pil_to_numpy(input_img_64)
@@ -137,14 +137,19 @@ def inpaint(input_img, input_img_with_mask, prompt):
     # Since gradio doesn't supply which pixels were drawn, we need to find it ourselves!
     # Assuming that all black pixels are meant for inpainting.
     # input_img_with_mask_64 = input_img_with_mask.convert('L').resize((64, 64), resample=Image.BICUBIC)
-    gray_scale_source_image = image_to_tensor(input_img_with_mask_64)
-    source_mask_64 = (gray_scale_source_image!=0).float()
+    gray_scale_source_image_64 = image_to_tensor(input_img_with_mask_64)
+    gray_scale_source_image_256 = image_to_tensor(input_img_with_mask_256)
+
+    source_mask_64 = (gray_scale_source_image_64!=0).float()
+    source_mask_256 = (gray_scale_source_image_256!=0).float()
+
     source_mask_64_img = tensor_to_image(source_mask_64)
 
     # The mask should always be a boolean 64x64 mask, and then we
     # can upsample it for the second stage.
     source_mask_64 = source_mask_64.unsqueeze(0)
-    source_mask_256 = F.interpolate(source_mask_64, (256, 256), mode='nearest')
+    source_mask_256 = source_mask_256.unsqueeze(0)
+    # source_mask_256 = F.interpolate(source_mask_64, (256, 256), mode='nearest')
 
 
     ##############################