Spaces:

sumit-ai-ml
/

SAM-with-YOLO

Runtime error

sumit-ai-ml commited on Oct 3, 2023

Commit

a5681c5

1 Parent(s): 37e59b4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -116,7 +116,8 @@ iface = gr.Interface(
             default="X-ray"
         )
     ],
-    outputs=gr.outputs.Image(type="numpy", label="Segmented Image"),
     title="YOLOv8 with SAM 😃",
     description='This software generates the segmentation mask for Aorta for Point of Care Ultrasound (POCUS) images'
 )

             default="X-ray"
         )
     ],
+    outputs=[gr.outputs.Image(type="numpy", label="Segmented Image"),
+             gr.outputs.Image(type="numpy", label="Segmentation Mask")],  # Add an output for the mask
     title="YOLOv8 with SAM 😃",
     description='This software generates the segmentation mask for Aorta for Point of Care Ultrasound (POCUS) images'
 )