Update app.py

app.py

@@ -111,7 +111,13 @@ def detect_clothing_onnx(image):
     categories = []
     
     for detection in detections:
-        x1, y1, x2, y2, conf, cls = detection[:6]  # 필요한 값만 추출
+        # detection에서 필요한 값 추출
+        x1, y1, x2, y2, conf, cls = detection[:6]
+        
+        # conf가 NumPy 배열인 경우, 단일 값으로 변환
+        if isinstance(conf, np.ndarray):
+            conf = conf.item()  # 배열을 단일 값으로 변환
+            
         if conf > 0.5:  # 신뢰도 임계값 설정
             category = onnx_model_labels[int(cls)]
             categories.append({
@@ -121,7 +127,6 @@ def detect_clothing_onnx(image):
             })
     
     return categories
-
 # 이미지 자르기
 def crop_image(image, bbox):
     return image.crop((bbox[0], bbox[1], bbox[2], bbox[3]))