victorisgeek commited on
Commit
50ffd5f
β€’
1 Parent(s): 73cf4be

Update nsfw_checker/opennsfw.py

Browse files
Files changed (1) hide show
  1. 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/yahoo/open_nsfw
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 True
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