Update app.py

app.py

@@ -255,18 +255,18 @@ with gr.Blocks(title="Wildlife Re-ID: Coarse-to-Fine Demo") as demo:
             # Candidate 1 Group
             with gr.Group():
                 log1 = gr.Markdown()
-                # FIX: Removed fixed height so it hugs the content
-                img1 = gr.Image(label="Visualization", show_download_button=False, show_label=False)
+                # FIX: Removed show_download_button AND height
+                img1 = gr.Image(label="Visualization", show_label=False)
             
             # Candidate 2 Group
             with gr.Group():
                 log2 = gr.Markdown()
-                img2 = gr.Image(label="Visualization", show_download_button=False, show_label=False)
+                img2 = gr.Image(label="Visualization", show_label=False)
                 
             # Candidate 3 Group
             with gr.Group():
                 log3 = gr.Markdown()
-                img3 = gr.Image(label="Visualization", show_download_button=False, show_label=False)
+                img3 = gr.Image(label="Visualization", show_label=False)
 
     submit_btn.click(
         fn=predict,