Spaces:
Running
on
T4
Running
on
T4
Vivien Chappelier
commited on
Commit
•
3e1fcf3
1
Parent(s):
481b311
try again
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ class BZHStableSignatureDemo(object):
|
|
| 51 |
self.pipe.vae.load_state_dict(sd, strict=False)
|
| 52 |
|
| 53 |
output = self.pipe(prompt, num_inference_steps=4, guidance_scale=0.0, output_type="pil")
|
| 54 |
-
return { "background": output.images[0] }
|
| 55 |
|
| 56 |
def attack_detect(self, img_edit, jpeg_compression, downscale, saturation):
|
| 57 |
|
|
|
|
| 51 |
self.pipe.vae.load_state_dict(sd, strict=False)
|
| 52 |
|
| 53 |
output = self.pipe(prompt, num_inference_steps=4, guidance_scale=0.0, output_type="pil")
|
| 54 |
+
return { "background": output.images[0], "layers": [], "composite": None }
|
| 55 |
|
| 56 |
def attack_detect(self, img_edit, jpeg_compression, downscale, saturation):
|
| 57 |
|