Spaces:
Running
Running
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()
|