remove useless tick

app.py

@@ -1,4 +1,3 @@
-# pip install -U gradio transformers pillow matplotlib
 
 import io
 from functools import lru_cache
@@ -15,7 +14,6 @@ MODELS = [
     "HuggingFaceM4/Idefics3-8B-Llama3",
 ]
 
-
 def _fig_to_pil(fig) -> Image.Image:
     buf = io.BytesIO()
     fig.savefig(buf, format="png", bbox_inches="tight", dpi=160)
@@ -26,7 +24,7 @@ def _fig_to_pil(fig) -> Image.Image:
 def get_viz(model_id: str) -> ImageVisualizer:
     return ImageVisualizer(model_id)
 
-def _run(model_id: str, image: Optional[Image.Image], use_sample: bool, add_grid: bool):
+def _run(model_id: str, image: Optional[Image.Image], add_grid: bool):
     viz = get_viz(model_id)
 
     captured = []
@@ -37,7 +35,8 @@ def _run(model_id: str, image: Optional[Image.Image], use_sample: bool, add_grid
 
     try:
         plt.show = _capture_show
-        viz.visualize(images=None if use_sample else image, add_grid=add_grid)
+        # if image is None, the visualizer will use its default sample
+        viz.visualize(images=image, add_grid=add_grid)
     finally:
         plt.show = orig_show
 
@@ -46,14 +45,11 @@ def _run(model_id: str, image: Optional[Image.Image], use_sample: bool, add_grid
     prompt_preview = viz.default_message(full_output=False)
     return left_img, right_img, prompt_preview
 
-def _resolve_and_run(model_pick, custom_model, image, use_sample, add_grid):
+def _resolve_and_run(model_pick, custom_model, image, add_grid):
     model_id = (custom_model or "").strip() or (model_pick or "").strip()
     if not model_id:
         raise gr.Error("Pick a model or enter one.")
-    return _run(model_id, image, use_sample, add_grid)
-
-def _on_image_change(_):
-    return False  # uncheck sample toggle
+    return _run(model_id, image, add_grid)
 
 def _preload_models():
     for mid in MODELS:
@@ -74,28 +70,22 @@ with gr.Blocks(title="Transformers Processor Visualizer", theme=theme) as demo:
         with gr.Column(scale=3):
             with gr.Row():
                 add_grid = gr.Checkbox(label="Show patch grid", value=True)
-                use_sample = gr.Checkbox(label="Use HF logo sample", value=True)
             image = gr.Image(label="Upload custom image", type="pil", height=140, sources=["upload"])
             gr.Markdown("## Output")
-
             with gr.Row():
                 left_output = gr.Image(label="Processor output", type="pil", height=900)
                 right_output = gr.Image(label="Global image (if any)", type="pil", height=900)
-
             prompt = gr.Textbox(label="Compact chat template preview", lines=2)
 
     # reactive updates
-    model_pick.change(_resolve_and_run, [model_pick, custom_model, image, use_sample, add_grid], [left_output, right_output, prompt])
-    custom_model.submit(_resolve_and_run, [model_pick, custom_model, image, use_sample, add_grid], [left_output, right_output, prompt])
-    add_grid.change(_resolve_and_run, [model_pick, custom_model, image, use_sample, add_grid], [left_output, right_output, prompt])
-    use_sample.change(_resolve_and_run, [model_pick, custom_model, image, use_sample, add_grid], [left_output, right_output, prompt])
-    image.change(_on_image_change, inputs=image, outputs=use_sample).then(
-        _resolve_and_run, [model_pick, custom_model, image, use_sample, add_grid], [left_output, right_output, prompt]
-    )
+    model_pick.change(_resolve_and_run, [model_pick, custom_model, image, add_grid], [left_output, right_output, prompt])
+    custom_model.submit(_resolve_and_run, [model_pick, custom_model, image, add_grid], [left_output, right_output, prompt])
+    add_grid.change(_resolve_and_run, [model_pick, custom_model, image, add_grid], [left_output, right_output, prompt])
+    image.change(_resolve_and_run, [model_pick, custom_model, image, add_grid], [left_output, right_output, prompt])
 
     # preload models into cache and render once
     demo.load(_preload_models, [], [])
-    demo.load(_resolve_and_run, [model_pick, custom_model, image, use_sample, add_grid], [left_output, right_output, prompt])
+    demo.load(_resolve_and_run, [model_pick, custom_model, image, add_grid], [left_output, right_output, prompt])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()