Initial Commit

app.py

@@ -35,13 +35,22 @@ def inference(image, input_label, model_choice):
                                                 point_labels=input_label,
                                                 multimask_output=True,
                                              )
-    mask = masks[0]
-    print(scores)
-    score = scores[0]
-    image2 = image.copy()
-    image2[mask, 0] = 255
-    image2[y-10:y+10, x-10:x+10, 2] = 255
-    return f"{score}", image2
+    mask1 = masks[0]
+    score1 = scores[0]
+    img1 = image.copy()
+    img1[mask1, 0] = 255
+    img1[y-10:y+10, x-10:x+10, 2] = 255
+    mask2 = masks[1]
+    score2 = scores[1]
+    img2 = image.copy()
+    img2[mask2, 0] = 255
+    img2[y-10:y+10, x-10:x+10, 2] = 255
+    mask3 = masks[2]
+    score3 = scores[2]
+    img3 = image.copy()
+    img3[mask3, 0] = 255
+    img3[y-10:y+10, x-10:x+10, 2] = 255
+    return f"{score1}", img1, f"{score2}", img2, f"{score3}", img3
 
 
 my_app = gr.Blocks()
@@ -55,13 +64,21 @@ with my_app:
                                           value='./images/truck.jpg').style(height=500, width=1000)
                 with gr.Row():
                     coords = gr.Label(label="Image Coordinate")
-                    with gr.Column():
-                        model_choice = gr.Dropdown(['ViT-base', 'ViT-large', 'ViT-huge'], label='Model Backbone')
-                        infer = gr.Button(label="Segment")
+                    model_choice = gr.Dropdown(['ViT-base', 'ViT-large', 'ViT-huge'], label='Model Backbone')
                 with gr.Row():
-                    score = gr.Label(label="Mask Confidence")
+                    infer = gr.Button(label="Segment")
                 with gr.Row():
-                    img_output = gr.Image(label="Output Mask").style(height=500, width=1000)
+                    score1 = gr.Label(label="Mask 1 Confidence")
+                with gr.Row():
+                    img_output1 = gr.Image(label="Output Mask 1").style(height=500, width=1000)
+                with gr.Row():
+                    score2 = gr.Label(label="Mask 2 Confidence")
+                with gr.Row():
+                    img_output2 = gr.Image(label="Output Mask 2").style(height=500, width=1000)
+                with gr.Row():
+                    score3 = gr.Label(label="Mask 3 Confidence")
+                with gr.Row():
+                    img_output3 = gr.Image(label="Output Mask 3").style(height=500, width=1000)
                     
         img_source.select(get_coords, [], coords)
         infer.click(
@@ -72,8 +89,12 @@ with my_app:
                 model_choice
             ],
             [
-                score,
-                img_output
+                score1,
+                img_output1,
+                score2,
+                img_output2,
+                score3,
+                img_output3, 
             ]
         )