Fixed some bugs

app.py

@@ -17,7 +17,7 @@ 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
+        gt_map = gt_map[:, :, 0] 
 
     # Check if the ground truth map is empty
     if np.sum(gt_map) == 0:
@@ -57,7 +57,7 @@ def process_image(image_input):
         outputs = model(**inputs, multimask_output=False)
 
     # Process model output
-    seg_prob = torch.sigmoid(outputs['pred_masks'].squeeze(1))
+    seg_prob = torch.sigmoid(outputs.pred_masks.squeeze(1))
     seg_prob = seg_prob.cpu().numpy().squeeze()
     seg = (seg_prob > 0.5).astype(np.uint8)