Spaces:

sumit-ai-ml
/

Aorta-segmentation

Sleeping

sumit-ai-ml commited on Nov 13, 2023

Commit

e919ebf

1 Parent(s): 4ac6b0b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,8 +39,8 @@ def segment_image(input_image):
 iface = gr.Interface(
     fn=segment_image,
-    inputs=gr.inputs.Image(type="pil", label="Upload an image"),
-    outputs=gr.outputs.Image(type="numpy", label="Segmented Image"),
     title="Aorta segmentation and Detection using YOLOv8 😃",
     description= 'This software generates the segementation mask for Aorta for the Point of Care Ultrasound (POCUS) images')

 iface = gr.Interface(
     fn=segment_image,
+    inputs=gr.components.Image(type="pil", label="Upload an image"),
+    outputs=gr.components.Image(type="numpy", label="Segmented Image"),
     title="Aorta segmentation and Detection using YOLOv8 😃",
     description= 'This software generates the segementation mask for Aorta for the Point of Care Ultrasound (POCUS) images')