Spaces:
Running
Running
Update src/backend.py
Browse files- 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:
|