Update tasks/ref_in.py

tasks/ref_in.py

@@ -69,9 +69,9 @@ def referring_inpainting(model, image, texts, inpainting_text, *args, **kwargs):
         width = image_crop.size[0]; height = image_crop.size[1]
         images_inpainting = pipe(prompt = inpainting_text.strip(), image=image_and_mask['image'], mask_image=image_and_mask['mask'], height=height, width=width).images[0]
         # put images_inpainting back to original image
-        image_ori.paste(images_inpainting)        
+        # image_ori.paste(images_inpainting)        
         torch.cuda.empty_cache()
-        return Image.fromarray(res) ,'' , image_ori
+        return Image.fromarray(res) ,'' , images_inpainting
     else:
         torch.cuda.empty_cache()
         return image_ori, 'text', Image.fromarray(res)