dzmu commited on
Commit
439fb23
·
verified ·
1 Parent(s): 31f4a97

Update src/backend.py

Browse files
Files changed (1) hide show
  1. src/backend.py +4 -1
src/backend.py CHANGED
@@ -1,4 +1,7 @@
1
- def analyze_outfit(input_img):
 
 
 
2
  # Handle both file paths and PIL Images
3
  if isinstance(input_img, str):
4
  try:
 
1
+ def analyze_outfit(input_img, yolo_person_model, yolo_fashion_model, clip_model, clip_preprocess,
2
+ all_prompts, style_prompts_end_index, FASHION_CLASSES, CATEGORY_LABEL_MAP,
3
+ response_templates, YOLO_PERSON_CONF_THRESHOLD, YOLO_FASHION_CONF_THRESHOLD,
4
+ YOLO_FASHION_HIGH_CONF_THRESHOLD, DEVICE):
5
  # Handle both file paths and PIL Images
6
  if isinstance(input_img, str):
7
  try: