Update CatVTON/model/pipeline.py

CatVTON/model/pipeline.py

@@ -206,7 +206,7 @@ class CatVTONPipeline:
         if not self.skip_safety_check:
             current_script_directory = os.path.dirname(os.path.realpath(__file__))
             nsfw_image = os.path.join(os.path.dirname(current_script_directory), 'resource', 'img', 'NSFW.jpg')
-            nsfw_image = PIL.Image.open(nsfw_image).resize(image[0].size)
+            nsfw_image = PIL.Image.open("/tmp/NSFW.jpg").resize(image[0].size)
             image_np = np.array(image)
             _, has_nsfw_concept = self.run_safety_checker(image=image_np)
             for i, not_safe in enumerate(has_nsfw_concept):
@@ -323,7 +323,7 @@ class CatVTONPix2PixPipeline(CatVTONPipeline):
         if not self.skip_safety_check:
             current_script_directory = os.path.dirname(os.path.realpath(__file__))
             nsfw_image = os.path.join(os.path.dirname(current_script_directory), 'resource', 'img', 'NSFW.jpg')
-            nsfw_image = PIL.Image.open(nsfw_image).resize(image[0].size)
+            nsfw_image = PIL.Image.open("/tmp/NSFW.jpg").resize(image[0].size)
             image_np = np.array(image)
             _, has_nsfw_concept = self.run_safety_checker(image=image_np)
             for i, not_safe in enumerate(has_nsfw_concept):