yasserrmd commited on
Commit
c5a0a3c
1 Parent(s): 4715e52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ pipe.to("cuda")
17
  image_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection",device=device)
18
  #text_classifier = pipeline("text-classification", model="eliasalbouzidi/distilbert-nsfw-text-classifier",device=device)
19
 
20
- NSFW_THRESHOLD = 0.5
21
 
22
  # Define the function to generate the sketch
23
  @spaces.GPU
 
17
  image_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection",device=device)
18
  #text_classifier = pipeline("text-classification", model="eliasalbouzidi/distilbert-nsfw-text-classifier",device=device)
19
 
20
+ NSFW_THRESHOLD = 0
21
 
22
  # Define the function to generate the sketch
23
  @spaces.GPU