update app

app.py

@@ -70,7 +70,7 @@ def segment_image(input_image):
 # --------------------
 # UI
 # --------------------
-with gr.Blocks(title="HemaScan Pro") as demo:
+with gr.Blocks(title="Malaria Cell Segmentation Tool") as demo:
     gr.Markdown("<h1 style='text-align:center; background:linear-gradient(90deg,#4facfe,#00f2fe); color:white; padding:10px;'>HemaScan Pro - Binary Segmentation</h1>")
 
     with gr.Accordion(" Documentation", open=False):
@@ -80,22 +80,25 @@ with gr.Blocks(title="HemaScan Pro") as demo:
         gr.Markdown("---")
         gr.Markdown(OUTPUT_EXPLANATION)
 
-    gr.Markdown("## Step 1: Upload Blood Smear Image")
     with gr.Row():
+        gr.Markdown("## Step 1: Upload Blood Smear Image")
         with gr.Column(scale=1):
             # Define Input component directly inside the column (No .render() needed)
             input_image = gr.Image(type="pil", label="Step 1: Upload Blood Smear Image", width=600, height=600) 
     
-    gr.Markdown("## Step 2: Click Submit for Segmentation")
-    with gr.Row():
-        submit_button = gr.Button("Submit for Segmentation", variant="primary")
-    gr.Markdown("## Output")
-    with gr.Row(scale=2):
-         # Define Output component directly inside the column (No .render() needed)
-        output_image = gr.Image(type="pil", label="Step 3: Predicted Masks", width=600, height=600)
+            gr.Markdown("## Step 2: Click Submit for Segmentation")
+            with gr.Row():
+                submit_button = gr.Button("Submit for Segmentation", variant="primary")
+        gr.Markdown("## Output")
+        with gr.Column(scale=1):
+             # Define Output component directly inside the column (No .render() needed)
+            output_image = gr.Image(type="pil", label="Step 3: Predicted Masks", width=600, height=600)
+
 
+    gr.markdown("---")
+    gr.markdown("## Example Images")
     gr.Examples(
-        examples=["data/1.jpg", "data/2.jpg", "data/3.jpg"],
+        examples=["data/1.png", "data/2.png", "data/3.png"],
         inputs=input_image,
         outputs=output_image,
         fn=segment_image,