Update app.py

app.py

@@ -243,7 +243,7 @@ def soft_nms(boxes, scores, sigma=0.5, thresh=0.001, method='gaussian'):
             
     # 刪除得分低於閾值的框
     keep = scores > thresh
-    boxes = boxes[keep]
+    boxes = boxes[keep.squeeze(-1)]  # 確保 keep 的維度與 boxes 匹配
     scores = scores[keep]
 
     return boxes, scores
@@ -265,7 +265,7 @@ async def detect_multiple_dogs(image, conf_threshold=0.35, iou_threshold=0.55):
         dogs.append((image, 1.0, [0, 0, image.width, image.height]))
     else:
         boxes = torch.stack(boxes)
-        scores = torch.stack(scores)
+        scores = torch.stack(scores).squeeze(-1)  # 確保 scores 是一維張量
         nms_boxes, nms_scores = soft_nms(boxes, scores, thresh=iou_threshold)
         
         for box, score in zip(nms_boxes, nms_scores):