update app.py

app.py

@@ -291,10 +291,16 @@ def run_grounded_sam(image_path, text_prompt, task_type, inpaint_prompt, box_thr
         mask = masks[0][0].cpu().numpy() # simply choose the first mask, which will be refine in the future release
         mask_pil = Image.fromarray(mask)
         image_pil = Image.fromarray(image)
+        # image_inpainting = sd_pipe(prompt=inpaint_prompt, image=image_pil, mask_image=mask_pil).images[0]
+
+        # resize for inpaint
+        image_source_for_inpaint = image_pil.resize((512, 512))
+        image_mask_for_inpaint = mask_pil.resize((512, 512))
+        image_inpainting = sd_pipe(prompt=inpaint_prompt, image=image_source_for_inpaint, mask_image=image_mask_for_inpaint).images[0]
+        image_inpainting = image_inpainting.resize((image_pil.size[0], image_pil.size[1]))
 
-        image = sd_pipe(prompt=inpaint_prompt, image=image_pil, mask_image=mask_pil).images[0]
         image_path = os.path.join(output_dir, f"grounded_sam_inpainting_output_{file_temp}.jpg")
-        image.save(image_path)
+        image_inpainting.save(image_path)
         image_result = cv2.cvtColor(cv2.imread(image_path), cv2.COLOR_BGR2RGB)
         os.remove(image_path)
         return image_result