Spaces:

yasserrmd
/

MagicDoodles

Running on Zero

yasserrmd commited on Oct 6

Commit

7efeba9

•

1 Parent(s): a960bc2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ pipe.to("cuda")
 # Load the NSFW classifier
 classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection",device=torch.device('cuda'))
-NSFW_THRESHOLD = 0.85
 # Define the function to generate the sketch
 @spaces.GPU

 # Load the NSFW classifier
 classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection",device=torch.device('cuda'))
+NSFW_THRESHOLD = 0.5
 # Define the function to generate the sketch
 @spaces.GPU