Update app.py

app.py

@@ -187,16 +187,20 @@ with gr.Blocks(title="Teeth Segmentation — Mask R-CNN") as demo:
         "The model will overlay colored masks on detected teeth."
     )
     
-    with gr.Row():
-        with gr.Column():
-            input_image = gr.Image(label="Input Image", type="pil",
-                examples=[
-                    ["example_image1.jfif"],
-                    ["example_image2.jfif"],
-                    ["example_image3.jfif"],
-                ])
-            analyze_btn = gr.Button("Analyze Image", variant="primary")
+    with gr.Column():
+        input_image = gr.Image(label="Input Image", type="pil")
+        analyze_btn = gr.Button("Analyze Image", variant="primary")
         output_image = gr.Image(label="Segmented Output", type="pil")
+
+        gr.Examples(
+            examples=[
+                ["example_image1.jfif"],
+                ["example_image2.jfif"],
+                ["example_image3.jfif"],
+            ],
+            inputs=input_image,
+        )
+
     
     analyze_btn.click(
         fn=run_inference,