output dataframe

app.py

@@ -23,10 +23,13 @@ def visualize_instance_seg_mask(mask, id2label):
 
     image = image / 255
 
-    label2count = {id2label[id]: id2count[id] / image_total_pixels for id in label_ids}
-    print(label2count)
+    dataframe = [[
+        f"{id2label[id]}",
+        f"{(100 * id2count[id] / image_total_pixels):.2f} %",
+        f"{np.sqrt(id2count[id] / image_total_pixels):.2f} m"
+        ] for id in label_ids if 'tree' in id2label[id]]
 
-    return image
+    return image, dataframe
 
 
 def query_image(img):
@@ -41,8 +44,8 @@ def query_image(img):
 demo = gr.Interface(
     query_image,
     inputs=[gr.Image()],
-    outputs="image",
-    title="maskformer-swin-large-coco results",
+    outputs=["image", gr.DataFrame(headers=None, default_value=None, label="Area Info")],
+    title="maskformer-swin-large-coco",
     allow_flagging="never",
     analytics_enabled=None
 )