Spaces:
Build error
Build error
AliA1997 commited on
Commit ·
4f566b3
1
Parent(s): eac19bc
fix classifier error
Browse files
app.py
CHANGED
|
@@ -6,7 +6,6 @@ from transformers import pipeline
|
|
| 6 |
|
| 7 |
from classify_utils import classify_image_if_nsfw, get_nsfw_classifier, get_normal_classifier
|
| 8 |
|
| 9 |
-
global classifier
|
| 10 |
|
| 11 |
def init_login():
|
| 12 |
try:
|
|
@@ -17,6 +16,7 @@ def init_login():
|
|
| 17 |
|
| 18 |
|
| 19 |
def init_classifier():
|
|
|
|
| 20 |
classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
|
| 21 |
|
| 22 |
|
|
|
|
| 6 |
|
| 7 |
from classify_utils import classify_image_if_nsfw, get_nsfw_classifier, get_normal_classifier
|
| 8 |
|
|
|
|
| 9 |
|
| 10 |
def init_login():
|
| 11 |
try:
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
def init_classifier():
|
| 19 |
+
global classifier
|
| 20 |
classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
|
| 21 |
|
| 22 |
|