dzmu commited on
Commit
6640bd7
·
verified ·
1 Parent(s): 1a91629

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()