sguna commited on
Commit
32cecd8
·
verified ·
1 Parent(s): 28c5c92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def classify_image(image):
44
  safe_prob = probs[0][0] # Safe probability
45
  unsafe_prob = probs[0][1] # Unsafe probability
46
  #safe_prob = sum(value if categories[i] in safe_categories else 0.0 for i, value in enumerate(probs[0]))
47
- #unsafe_prob = sum(value if categories[i] in unsafe_categoriessafe_categories else 0.0 for i, value in enumerate(probs[0]))
48
 
49
  #debug
50
  for i, value in enumerate(probs[0]):
 
44
  safe_prob = probs[0][0] # Safe probability
45
  unsafe_prob = probs[0][1] # Unsafe probability
46
  #safe_prob = sum(value if categories[i] in safe_categories else 0.0 for i, value in enumerate(probs[0]))
47
+ #unsafe_prob = sum(value if categories[i] in unsafe_categories else 0.0 for i, value in enumerate(probs[0]))
48
 
49
  #debug
50
  for i, value in enumerate(probs[0]):