minor(ui)

app.py

@@ -264,12 +264,13 @@ with gr.Blocks() as iface:
                 predict_button = gr.Button("Predict")
                 augment_button = gr.Button("Augment & Predict")
                 image_output = gr.Image(label="Processed Image", visible=False)
-                forensics_gallery = gr.Gallery(label="Post Processed Images", visible=True)
 
 
             with gr.Column(scale=2):
                 # Custom HTML component to display results in 5 columns
                 results_html = gr.HTML(label="Model Predictions")
+                forensics_gallery = gr.Gallery(label="Post Processed Images", visible=True)
+
                 outputs = [image_output, forensics_gallery, results_html]
         
         # Show/hide rotate slider based on selected augmentation method