Mridul2003 commited on
Commit
4e928c3
·
verified ·
1 Parent(s): ed4fde3

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +3 -3
api.py CHANGED
@@ -148,16 +148,16 @@ def filter_comment():
148
 
149
  # Compose annotation message
150
  if not result.get("text"):
151
- message = "Toxic content is present\nImage is blurred"
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 = f"The image contains {max_label} ({filtered[max_label]:.2f}) content"
159
  else:
160
- message = "Toxic content is present\nImage is blurred"
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()