Upload app.py

app.py

@@ -447,10 +447,12 @@ def predict(image_input, gen_threshold, char_threshold, output_mode):
     output_text = ", ".join(output_tags)
 
     if output_mode == "Tags Only":
-        return output_text, None
+        # Return the text and an update to make the visualization invisible
+        return output_text, gr.update(value=None, visible=False)
     else: # Visualization
         viz_image = visualize_predictions(original_pil_image, predictions, gen_threshold)
-        return output_text, viz_image
+        # Return the text and an update to show the visualization
+        return output_text, gr.update(value=viz_image, visible=True)
 
 # --- Gradio Interface Definition ---
 import time
@@ -563,10 +565,10 @@ with gr.Blocks(css=css, js=js) as demo:
     # Examples
     gr.Examples(
         examples=[
-            ["https://pbs.twimg.com/media/GpiBUQZawAAetgr.jpg", 0.55, 0.5, "Tags + Visualization"],
-            ["https://pbs.twimg.com/media/GooBBQHWcAAJj2q.jpg", 0.5, 0.5, "Tags Only"],
-            ["https://m.media-amazon.com/images/I/61FwAqFu4PL.jpg", 0.55, 0.5, "Tags + Visualization"],
-            ["https://cdn.donmai.us/sample/5d/ad/__kanae_and_kanae_nijisanji_drawn_by_cococall__sample-5dadca17680ef18c18daaf75507c4b12.jpg", 0.45, 0.45, "Tags + Visualization"]
+            ["https://pbs.twimg.com/media/GXBXsRvbQAAg1kp.jpg", 0.55, 0.5, "Tags + Visualization"],
+            ["https://pbs.twimg.com/media/GjlX0gibcAA4EJ4.jpg", 0.5, 0.5, "Tags Only"],
+            ["https://pbs.twimg.com/media/Gj4nQbjbEAATeoH.jpgg", 0.55, 0.5, "Tags + Visualization"],
+            ["https://pbs.twimg.com/media/GkbtX0GaoAMlUZt.jpg", 0.45, 0.45, "Tags + Visualization"]
         ],
         inputs=[image_input, gen_threshold, char_threshold, output_mode],
         outputs=[output_tags, output_visualization],