Update represent result

app.py

@@ -18,7 +18,7 @@ with gr.Blocks() as demo:
 
     # Advanced parameters for Detailed Analysis
     with gr.Row():
-        conf_threshold = gr.Slider(0, 1, value=0.3, label="Confidence Threshold")
+        conf_threshold = gr.Slider(0, 1, value=0.2, label="Confidence Threshold")
         iou_threshold = gr.Slider(0, 1, value=0.45, label="Overlap Threshold")
         label_mode = gr.Dropdown(
             ["Draw box", "Draw Label", "Draw Confidence", "Censor Predictions"],

utils/data_processing.py

@@ -148,7 +148,7 @@ def detect_nsfw(input_image, conf_threshold=0.3, iou_threshold=0.45, label_mode=
     )
 
     # Construct detailed result text
-    if "head" in label_counts:
+    if "head" in label_counts and len(label_counts) == 1:
         head_count = label_counts["head"]
         result_text = (
             f"Detected content:\n - head: {head_count} instance(s)\nThis image appears to be safe."