Update app.py

app.py

@@ -17,13 +17,20 @@ def predict(image):
     mask = (mask[0] > 0.5).astype(np.uint8) * 255 
     mask = cv2.resize(mask, (original_width, original_height)) 
     return mask 
+
+with gr.Blocks() as iface:
+    gr.Markdown("# Tooth Segmentation Model")
+    gr.Markdown("Upload a dental X-ray image to generate the annotation mask.")
     
-iface = gr.Interface( 
-    fn=predict, 
-    inputs=gr.Image(type="numpy", label="Input X-ray Image"), 
-    outputs=gr.Image(type="numpy", label="Annotation Mask"), 
-    title="Tooth Segmentation Model", 
-    description="Upload a dental X-ray image to generate the annotation mask.", ) 
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="numpy", label="Input X-ray Image")
+            submit_button = gr.Button("Predict")
+
+        with gr.Column():
+            output_image = gr.Image(type="numpy", label="Annotation Mask")
+
+    submit_button.click(fn=predict, inputs=input_image, outputs=output_image)
 
 if __name__ == "__main__": 
     iface.launch(share=True)