risunobushi commited on
Commit
f1c6fd2
Β·
1 Parent(s): f877b6a

add dress functionalities

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,6 +41,7 @@ REQUEST_TIMEOUT = int(os.getenv("SUPABASE_FN_TIMEOUT", "240")) # seconds
41
  WORKFLOW_CHOICES = [
42
  "eyewear",
43
  "footwear",
 
44
  ]
45
 
46
  if not SUPABASE_URL or not SUPABASE_SECRET_KEY:
@@ -277,7 +278,7 @@ def generate(base_img: Image.Image, garment_img: Image.Image, workflow_choice: s
277
  # Gradio UI
278
  # -----------------------------------------------------------------------------
279
 
280
- description = "Upload a person photo (Base) and a product image. Select between Eyewear and Footwear to switch between the two available models. Click πŸ‘‰ **Generate** to try on a product." # noqa: E501
281
 
282
  with gr.Blocks(title="YOURMIRROR.IO - SM4LL-VTON Demo") as demo:
283
  # Header
 
41
  WORKFLOW_CHOICES = [
42
  "eyewear",
43
  "footwear",
44
+ "dress",
45
  ]
46
 
47
  if not SUPABASE_URL or not SUPABASE_SECRET_KEY:
 
278
  # Gradio UI
279
  # -----------------------------------------------------------------------------
280
 
281
+ description = "Upload a person photo (Base) and a product image. Select between Eyewear, Footwear, or Full-Body Garments to switch between the three available models. Click πŸ‘‰ **Generate** to try on a product." # noqa: E501
282
 
283
  with gr.Blocks(title="YOURMIRROR.IO - SM4LL-VTON Demo") as demo:
284
  # Header