Update app.py

app.py

@@ -57,19 +57,16 @@ def predict(image):
     return mask_final
 
 # إنشاء واجهة Gradio
-with gr.Blocks() as iface:
-    gr.Markdown("# Tooth Segmentation Model")
-    gr.Markdown("Upload a dental X-ray image to generate the annotation mask.")
-    
-    with gr.Row():
-        with gr.Column():
-            input_image = gr.File(label="Input X-ray Image")
-            submit_button = gr.Button("Predict")
-
-        with gr.Column():
-            output_image = gr.File(label="Annotation Mask")
-
+with gr.Blocks() as iface: 
+    gr.Markdown("# Tooth Segmentation Model") 
+    gr.Markdown("Upload a dental X-ray image to generate the annotation mask.") 
+    with gr.Row(): 
+        with gr.Column(): 
+            input_image = gr.Image(label="Input X-ray Image", type="numpy") 
+            submit_button = gr.Button("Predict") 
+        with gr.Column(): 
+            output_image = gr.Image(label="Annotation Mask", type="numpy") 
+            
     submit_button.click(fn=predict, inputs=input_image, outputs=output_image)
-
 if __name__ == "__main__":
     iface.launch()