Spaces:
Sleeping
Sleeping
pengdaqian
commited on
Commit
•
7f37981
1
Parent(s):
b2502f7
fix now
Browse files- img_nsfw.py +1 -1
img_nsfw.py
CHANGED
@@ -87,4 +87,4 @@ def check_nsfw(img, pipe):
|
|
87 |
from torch.cuda.amp import autocast
|
88 |
with autocast():
|
89 |
_, nsfw_tags = pipe.safety_checker.forward(clip_input=safety_checker_input.pixel_values, images=img)
|
90 |
-
return nsfw_tags
|
|
|
87 |
from torch.cuda.amp import autocast
|
88 |
with autocast():
|
89 |
_, nsfw_tags = pipe.safety_checker.forward(clip_input=safety_checker_input.pixel_values, images=img)
|
90 |
+
return nsfw_tags['bad_tags']
|