da_nsfw_checker

Running

yoinked commited on Apr 5, 2024

Commit

453c220

verified ·

1 Parent(s): 4700115

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -75,6 +75,7 @@ class WaifuDiffusionInterrogator:
         _, height, _, _ = self._model.get_inputs()[0].shape
         # alpha to white
         image = image.convert('RGBA')
         new_image = Image.new('RGBA', image.size, 'WHITE')
         new_image.paste(image, mask=image)

         _, height, _, _ = self._model.get_inputs()[0].shape
         # alpha to white
+        print(image)
         image = image.convert('RGBA')
         new_image = Image.new('RGBA', image.size, 'WHITE')
         new_image.paste(image, mask=image)