minor: ui

app.py

@@ -282,8 +282,9 @@ def predict_image_with_html(img, confidence_threshold, augment_methods, rotate_d
 
     # Second pass - enhanced visibility
     ela2 = ELA(img_np, quality=75, scale=75, contrast=25, linear=False, grayscale=True)
+    ela3 = ELA(img_np, quality=75, scale=75, contrast=25, linear=False, grayscale=False)
     
-    forensics_images = [img_pil, ela1, ela2, gradient_image, minmax_image]
+    forensics_images = [img_pil, ela1, ela2, ela3, gradient_image, minmax_image]
     
     html_content = generate_results_html(results)
     return img_pil, forensics_images, html_content
@@ -319,7 +320,7 @@ with gr.Blocks(css="#post-gallery { overflow: hidden !important;} .grid-wrap{ ov
                     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, columns=[5], rows=[1], container=False, height="auto", object_fit="contain", elem_id="post-gallery")
+                        forensics_gallery = gr.Gallery(label="Post Processed Images", visible=True, columns=[3], rows=[2], container=False, height="auto", object_fit="contain", elem_id="post-gallery")
 
                         outputs = [image_output, forensics_gallery, results_html]