Update handler.py

handler.py

@@ -145,10 +145,10 @@ class EndpointHandler:
         print(f"Confidences shape: {confidences.shape}, Boxes shape: {boxes.shape}")
 
         face_info_list = []
-        for i in range(boxes.shape[0]):
+        for i in range(len(boxes)):
             box = boxes[i]
             conf = confidences[i]
-            if conf.shape[0] > 0 and conf[0] > 0.7:  # Fixing the out-of-bounds issue
+            if conf[0] > 0.7:  # Fixing the out-of-bounds issue
                 x1, y1, x2, y2 = box[0] * 320, box[1] * 240, box[2] * 320, box[3] * 240
                 face_info_list.append({"bbox": [x1, y1, x2, y2]})
         return face_info_list
@@ -271,4 +271,3 @@ class EndpointHandler:
     def apply_style(self, style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
         p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
         return p.replace("{prompt}", positive), n + " " + negative
-