Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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]):
|