Spaces:
Running
Running
victorisgeek
commited on
Commit
•
50ffd5f
1
Parent(s):
73cf4be
Update nsfw_checker/opennsfw.py
Browse files- nsfw_checker/opennsfw.py +2 -2
nsfw_checker/opennsfw.py
CHANGED
@@ -5,7 +5,7 @@ import onnxruntime
|
|
5 |
import numpy as np
|
6 |
from tqdm import tqdm
|
7 |
|
8 |
-
# https://github.com/
|
9 |
|
10 |
class NSFWChecker:
|
11 |
def __init__(self, model_path=None, providers=["CPUExecutionProvider"]):
|
@@ -33,5 +33,5 @@ class NSFWChecker:
|
|
33 |
|
34 |
if score > threshold:
|
35 |
print(f"Detected nsfw score:{score}")
|
36 |
-
return
|
37 |
return False
|
|
|
5 |
import numpy as np
|
6 |
from tqdm import tqdm
|
7 |
|
8 |
+
# https://github.com/chemistzombie/stable-diffusion-unfiltered.git
|
9 |
|
10 |
class NSFWChecker:
|
11 |
def __init__(self, model_path=None, providers=["CPUExecutionProvider"]):
|
|
|
33 |
|
34 |
if score > threshold:
|
35 |
print(f"Detected nsfw score:{score}")
|
36 |
+
return False
|
37 |
return False
|