fix tab and sort mask by size

app.py

@@ -41,8 +41,9 @@ with block:
         )
         with gr.Row():
             with gr.Column():
-                with gr.Tab():
+                with gr.Tab("Upload Image"):
                     input_image = gr.Image(label='Input', type='pil', tool=None) # mirror_webcam = False
+                with gr.Tab("Webcam"):
                     input_image = gr.Image(label='Input', type='pil', tool=None, source="webcam") # mirror_webcam = False
                 with gr.Row():
                     sam_encode_btn = gr.Button('Encode', variant='primary')

inference.py

@@ -184,18 +184,15 @@ class SegmentPredictor:
     def cond_pred(self, pts, lbls):
         lbls = np.array(lbls)
         pts = np.array(pts)
-        
-        print(pts)
-        print(lbls)
-
         masks, _, _ = self.conditioned_pred.predict(
             point_coords=pts,
             point_labels=lbls,
             multimask_output=True
             )
+        idxs = np.argsort(masks.sum(axis=(1,2)))
         sam_masks = []
-        for i,mask in enumerate(masks):
-            sam_masks.append((mask, str(i)))
+        for n,i in enumerate(idxs):
+            sam_masks.append((masks[i], str(n)))
         return sam_masks