dzmu commited on
Commit
1cd8cba
·
verified ·
1 Parent(s): 8de791b

Update src/backend.py

Browse files
Files changed (1) hide show
  1. src/backend.py +11 -1
src/backend.py CHANGED
@@ -1,5 +1,15 @@
1
-
 
 
2
  import numpy as np
 
 
 
 
 
 
 
 
3
 
4
  def analyze_outfit(input_img, yolo_person_model, yolo_fashion_model, clip_model, clip_preprocess,
5
  all_prompts, style_prompts_end_index, FASHION_CLASSES, CATEGORY_LABEL_MAP,
 
1
+ import gradio as gr
2
+ import torch
3
+ import clip
4
  import numpy as np
5
+ import random
6
+ import os
7
+ from PIL import Image
8
+ from ultralytics import YOLO # Needed for both person and fashion detection
9
+ from gtts import gTTS
10
+ import uuid
11
+ import time
12
+ import tempfile
13
 
14
  def analyze_outfit(input_img, yolo_person_model, yolo_fashion_model, clip_model, clip_preprocess,
15
  all_prompts, style_prompts_end_index, FASHION_CLASSES, CATEGORY_LABEL_MAP,