add img examples

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,3 @@
+.idea/
+__pycache__/
+src/models/__pycache__/

app.py

@@ -17,7 +17,6 @@ try:
 except Exception:
     pass
 
-
 APP_TITLE = "Content Moderation Demo (Image & Video)"
 APP_DESC = """
 Minimal prototype: image/video analysis, model & category selection, and threshold control.
@@ -25,6 +24,13 @@ Minimal prototype: image/video analysis, model & category selection, and thresho
 
 MODEL_NAMES = list(REGISTRY.keys())
 
+IMG_EXAMPLES = [
+    # [model, image_path, categories, threshold]
+    ["clip-multilabel", "examples/gambling_alcohol.jpg", ALL_CATEGORIES, 0.50],
+    ["wdeva02-multilabel", "examples/smoke_alcohol.jpg", ALL_CATEGORIES, 0.50],
+    ["animetimm-multilabel", "examples/gambling_smoke_alcohol.jpg", ALL_CATEGORIES, 0.50],
+]
+
 # ---------- Shared ----------
 def on_model_change(model_name):
     if model_name in NUDENET_ONLY:
@@ -97,11 +103,6 @@ def analyze_video(model_name, video_file, selected_categories, threshold, sampli
 
         rows = []
         for seg_id, (a, b) in enumerate(union_runs, start=1):
-            for i in range(a, b + 1):
-                st = frame_stats.get(i)
-                if not st:
-                    continue
-
             cat_counts = {c: 0 for c in req}
             cat_maxp   = {c: 0.0 for c in req}
             for i in range(a, b + 1):
@@ -141,7 +142,6 @@ def analyze_video(model_name, video_file, selected_categories, threshold, sampli
 
         return df, out_video
 
-
 # ---------- UI ----------
 with gr.Blocks(title=APP_TITLE, css=".wrap-row { gap: 16px; }") as demo:
     gr.Markdown(f"# {APP_TITLE}")
@@ -166,6 +166,16 @@ with gr.Blocks(title=APP_TITLE, css=".wrap-row { gap: 16px; }") as demo:
             btn.click(analyze_image, inputs=[model_dd, inp_img, categories, threshold],
                       outputs=[verdict, scores_df, extra_tags])
 
+            gr.Examples(
+                label="Try an example (Image)",
+                examples=IMG_EXAMPLES,
+                inputs=[model_dd, inp_img, categories, threshold],
+                outputs=[verdict, scores_df, extra_tags],
+                fn=analyze_image,
+                run_on_click=True,
+                cache_examples=False,
+            )
+
         with gr.Tab("Video"):
             with gr.Row(elem_classes=["wrap-row"]):
                 with gr.Column(scale=1, min_width=360):