Initial Commit

app.py

@@ -36,10 +36,11 @@ def inference(image, input_label, model_choice):
                                                 multimask_output=True,
                                              )
     mask = masks[0]
+    score = scores[0]
     image2 = image.copy()
     image2[mask, 0] = 255
     image2[y-10:y+10, x-10:x+10, 2] = 255
-    return image2
+    return score, image2
 
 
 my_app = gr.Blocks()
@@ -55,6 +56,7 @@ with my_app:
                     model_choice = gr.Dropdown(['ViT-base', 'ViT-large', 'ViT-huge'], label='Model Backbone')
                     infer = gr.Button(label="Segment")
                 with gr.Column():
+                    score = gr.Label(label="Mask Confidence")
                     img_output = gr.Image(label="Output Mask", shape=(1024, 1024))
                     
         img_source.select(get_coords, [], coords)
@@ -66,6 +68,7 @@ with my_app:
                 model_choice
             ],
             [
+                score,
                 img_output
             ]
         )