Spaces:

imatag
/

stable-signature-bzh

Sleeping

Vivien Chappelier commited on Nov 30, 2023

Commit

3e1fcf3

•

1 Parent(s): 481b311

try again

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ class BZHStableSignatureDemo(object):
         self.pipe.vae.load_state_dict(sd, strict=False)
         output = self.pipe(prompt, num_inference_steps=4, guidance_scale=0.0, output_type="pil")
-        return { "background": output.images[0] }
     def attack_detect(self, img_edit, jpeg_compression, downscale, saturation):

         self.pipe.vae.load_state_dict(sd, strict=False)
         output = self.pipe(prompt, num_inference_steps=4, guidance_scale=0.0, output_type="pil")
+        return { "background": output.images[0], "layers": [], "composite": None }
     def attack_detect(self, img_edit, jpeg_compression, downscale, saturation):