Boboiazumi commited on
Commit
cf86522
·
verified ·
1 Parent(s): 3e68df0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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'Boneka ini terlalu seksi dan tidak aman dilihat anak kecil (NSFW) [{label[predicted_class]}:{score}]'
34
  else:
35
- output = f'Boneka ini aman (SAFE) [{label[predicted_class]}:{score}]'
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