yasserrmd commited on
Commit
c66e22e
1 Parent(s): 0a4d4a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ pipe.to("cuda")
16
  # Load the NSFW classifier
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
- classifier("I see you’ve set aside this special time to humiliate yourself in public.")
20
  NSFW_THRESHOLD = 0.5
21
 
22
  # Define the function to generate the sketch
 
16
  # Load the NSFW classifier
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