Update tasks/ref_in.py

tasks/ref_in.py

@@ -37,7 +37,7 @@ def crop_image(input_image):
 def referring_inpainting(model, image, texts, inpainting_text, *args, **kwargs):
     model.model.metadata = metadata
     texts = [[texts if texts.strip().endswith('.') else (texts.strip() + '.')]]
-    image_ori = transform(image)
+    image_ori = crop_image(transform(image))
 
     with torch.no_grad():
         width = image_ori.size[0]
@@ -58,10 +58,10 @@ def referring_inpainting(model, image, texts, inpainting_text, *args, **kwargs):
     
     if inpainting_text not in ['no', '']:
         # if we want to do inpainting
-        image_crop = crop_image(image_ori)
+        image_crop = image_ori
         struct2 = ndimage.generate_binary_structure(2, 2)
         mask_dilated = ndimage.binary_dilation(grd_mask[0], structure=struct2, iterations=3).astype(grd_mask[0].dtype)
-        mask = crop_image(Image.fromarray(mask_dilated * 255).convert('RGB'))
+        mask = Image.fromarray(mask_dilated * 255).convert('RGB')
         image_and_mask = {
             "image": image_crop,
             "mask": mask,