yasserrmd commited on
Commit
85913ad
1 Parent(s): 3654a3e

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
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.3
21
 
22
  # Define the function to generate the sketch
23
  @spaces.GPU