Update app.py

app.py

@@ -113,6 +113,14 @@ def find_similar_images(query_embedding, top_k=5):
         })
     return results
 
+# 먼저 라벨 리스트를 정의합니다
+labels = [
+    "bag", "coat", "dress", "fabric", "hat", "Hoodie", "jacket", "Jacket",
+    "jean", "m2m", "Mid-lenght dress", "Pants", "plain", "shirt", "Shirt",
+    "shoe", "short", "shorts", "skirt", "slacks", "suit", "sunglass",
+    "sweat", "tie", "tracksuit", "tshirt"
+]
+
 def detect_clothing(image):
     results = yolo_model(image)
     detections = results[0].boxes.data.cpu().numpy()
@@ -120,7 +128,7 @@ def detect_clothing(image):
     for detection in detections:
         x1, y1, x2, y2, conf, cls = detection
         category = yolo_model.names[int(cls)]
-        if category in labels = ["bag","coat","dress","fabric","hat","Hoodie","jacket","Jacket","jean","m2m","Mid-lenght dress","Pants","plain","shirt","Shirt","shoe","short","shorts","skirt","slacks","suit","sunglass","sweat","tie","tracksuit","tshirt"]:
+        if category in labels:
             categories.append({
                 'category': category,
                 'bbox': [int(x1), int(y1), int(x2), int(y2)],