fixed some bugs

app.py

@@ -15,6 +15,10 @@ model = SamModel.from_pretrained("./checkpoint",local_files_only=True)
 
 def get_bbox(gt_map):
     
+    if gt_map.ndim > 2:
+        gt_map = gt_map[:, :, 0]  # Assuming the mask is the same across all channels
+
+    # Check if the ground truth map is empty
     if np.sum(gt_map) == 0:
         return [0, 0, gt_map.shape[1], gt_map.shape[0]]