Update app.py

app.py

@@ -253,23 +253,22 @@ with gr.Blocks(title=title) as demo:
 
     gr.Markdown("## Test Samples")
 
-    with gr.Row(scale=1):
-        with gr.Column(scale=1):
-            gr.Image("images/NORMAL.jpeg", label="Image")
-        with gr.Column():
-            gr.Markdown("**NORMAL**")
+    with gr.Row():
 
-    with gr.Row(scale=1):
+        # NORMAL
         with gr.Column(scale=1):
-            gr.Image("images/VIRAL.jpeg", label="Image")
-        with gr.Column():
-            gr.Markdown("**VIRAL**")
-
-    with gr.Row(scale=1):
+            gr.Markdown("### NORMAL")
+            gr.Image("images/NORMAL.jpeg", show_label=False, height=220)
+    
+        # VIRAL
         with gr.Column(scale=1):
-            gr.Image("images/BACT.jpeg", label="Image")
-        with gr.Column():
-            gr.Markdown("**BACTERIAL**")
+            gr.Markdown("### VIRAL")
+            gr.Image("images/VIRAL.jpeg", show_label=False, height=220)
+    
+        # BACTERIAL
+        with gr.Column(scale=1):
+            gr.Markdown("### BACTERIAL")
+            gr.Image("images/BACT.jpeg", show_label=False, height=220)
 
 if __name__ == "__main__":
     demo.launch()