more accurate masking

app.py

@@ -129,7 +129,7 @@ def inpaint(input_img, prompt):
 
     input_img_with_mask_64 = input_img_with_mask_converted.resize((64, 64), resample=Image.BICUBIC)
     # TODO: make 256x256 mask more accurate when upscaling?
-    # input_img_with_mask_256 = input_img_with_mask_converted.resize((256, 256), 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)
@@ -139,18 +139,18 @@ def inpaint(input_img, prompt):
     # 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_64 = image_to_tensor(input_img_with_mask_64)
-    # gray_scale_source_image_256 = image_to_tensor(input_img_with_mask_256)
+    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_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 = source_mask_256.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')
 
 
     ##############################