Spaces:
Running
on
T4
Running
on
T4
Vivien Chappelier
commited on
Commit
•
4dfce87
1
Parent(s):
3e1fcf3
almost there..
Browse files
app.py
CHANGED
|
@@ -56,6 +56,7 @@ class BZHStableSignatureDemo(object):
|
|
| 56 |
def attack_detect(self, img_edit, jpeg_compression, downscale, saturation):
|
| 57 |
|
| 58 |
img = img_edit["composite"]
|
|
|
|
| 59 |
# attack
|
| 60 |
if downscale != 1:
|
| 61 |
size = img.size
|
|
|
|
| 56 |
def attack_detect(self, img_edit, jpeg_compression, downscale, saturation):
|
| 57 |
|
| 58 |
img = img_edit["composite"]
|
| 59 |
+
img = img.convert("RGB")
|
| 60 |
# attack
|
| 61 |
if downscale != 1:
|
| 62 |
size = img.size
|