Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -30,9 +30,9 @@ def inference(image):
|
|
30 |
score = output[0][predicted_class]
|
31 |
|
32 |
if label[predicted_class] == "nsfw":
|
33 |
-
output = f'
|
34 |
else:
|
35 |
-
output = f'
|
36 |
|
37 |
return output
|
38 |
|
|
|
30 |
score = output[0][predicted_class]
|
31 |
|
32 |
if label[predicted_class] == "nsfw":
|
33 |
+
output = f'NSFW [{label[predicted_class]}:{score}]'
|
34 |
else:
|
35 |
+
output = f'SAFE [{label[predicted_class]}:{score}]'
|
36 |
|
37 |
return output
|
38 |
|