Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ pipe.to("cuda")
|
|
| 12 |
|
| 13 |
# Load the NSFW classifier
|
| 14 |
classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection",device=torch.device('cuda'))
|
| 15 |
-
NSFW_THRESHOLD = 0.
|
| 16 |
|
| 17 |
# Define the function to generate the sketch
|
| 18 |
@spaces.GPU
|
|
|
|
| 12 |
|
| 13 |
# Load the NSFW classifier
|
| 14 |
classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection",device=torch.device('cuda'))
|
| 15 |
+
NSFW_THRESHOLD = 0.5
|
| 16 |
|
| 17 |
# Define the function to generate the sketch
|
| 18 |
@spaces.GPU
|