Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
|
@@ -148,16 +148,16 @@ def filter_comment():
|
|
| 148 |
|
| 149 |
# Compose annotation message
|
| 150 |
if not result.get("text"):
|
| 151 |
-
|
| 152 |
else:
|
| 153 |
toxic_result = result.get("toxic_result", {})
|
| 154 |
exclude_keys = {"safe", "identity_hate_custom", "not_identity_hate_custom"}
|
| 155 |
filtered = {k: v for k, v in toxic_result.items() if k not in exclude_keys}
|
| 156 |
if filtered:
|
| 157 |
max_label = max(filtered, key=filtered.get)
|
| 158 |
-
message =
|
| 159 |
else:
|
| 160 |
-
message = "
|
| 161 |
|
| 162 |
# Encode blurred image to base64
|
| 163 |
buffer = io.BytesIO()
|
|
|
|
| 148 |
|
| 149 |
# Compose annotation message
|
| 150 |
if not result.get("text"):
|
| 151 |
+
message = "The image contains some toxic content"
|
| 152 |
else:
|
| 153 |
toxic_result = result.get("toxic_result", {})
|
| 154 |
exclude_keys = {"safe", "identity_hate_custom", "not_identity_hate_custom"}
|
| 155 |
filtered = {k: v for k, v in toxic_result.items() if k not in exclude_keys}
|
| 156 |
if filtered:
|
| 157 |
max_label = max(filtered, key=filtered.get)
|
| 158 |
+
message = "The image contains some toxic content"
|
| 159 |
else:
|
| 160 |
+
message = "The image contains some toxic content"
|
| 161 |
|
| 162 |
# Encode blurred image to base64
|
| 163 |
buffer = io.BytesIO()
|