Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -139,7 +139,7 @@ def analyze_outfit(input_img: Image.Image):
|
|
| 139 |
# 2) YOLO Fashion Detection (on the cropped image)
|
| 140 |
detected_clothing_item = "fit" # Default item if no clothing detected
|
| 141 |
try:
|
| 142 |
-
fashion_results = fashion_model(cropped_img, verbose=False)
|
| 143 |
if len(fashion_results[0].boxes) > 0:
|
| 144 |
fashion_boxes = fashion_results[0].boxes.xyxy.cpu().numpy()
|
| 145 |
fashion_classes = fashion_results[0].boxes.cls.cpu().numpy()
|
|
|
|
| 139 |
# 2) YOLO Fashion Detection (on the cropped image)
|
| 140 |
detected_clothing_item = "fit" # Default item if no clothing detected
|
| 141 |
try:
|
| 142 |
+
fashion_results = fashion_model(cropped_img, conf=0.1, verbose=False)
|
| 143 |
if len(fashion_results[0].boxes) > 0:
|
| 144 |
fashion_boxes = fashion_results[0].boxes.xyxy.cpu().numpy()
|
| 145 |
fashion_classes = fashion_results[0].boxes.cls.cpu().numpy()
|