mrfreak72 commited on
Commit
a240c62
Β·
verified Β·
1 Parent(s): 810e590
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -40,7 +40,6 @@ def pil_to_binary_mask(pil_image, threshold=0):
40
  return output_mask
41
 
42
 
43
- base_path = 'yisol/IDM-VTON'
44
  example_path = os.path.join(os.path.dirname(__file__), 'example')
45
 
46
  unet = UNet2DConditionModel.from_pretrained(
@@ -260,8 +259,8 @@ for ex_human in human_list_path:
260
 
261
  image_blocks = gr.Blocks().queue()
262
  with image_blocks as demo:
263
- gr.Markdown("## IDM-VTON πŸ‘•πŸ‘”πŸ‘š")
264
- gr.Markdown("Virtual Try-on with your image and garment image. Check out the [source codes](https://github.com/yisol/IDM-VTON) and the [model](https://huggingface.co/yisol/IDM-VTON)")
265
  with gr.Row():
266
  with gr.Column():
267
  imgs = gr.ImageEditor(sources='upload', type="pil", label='Human. Mask with pen or use auto-masking', interactive=True)
 
40
  return output_mask
41
 
42
 
 
43
  example_path = os.path.join(os.path.dirname(__file__), 'example')
44
 
45
  unet = UNet2DConditionModel.from_pretrained(
 
259
 
260
  image_blocks = gr.Blocks().queue()
261
  with image_blocks as demo:
262
+ gr.Markdown("## DRESSIFY.TECH")
263
+ gr.Markdown("Virtual Try-on for Clothing brands")
264
  with gr.Row():
265
  with gr.Column():
266
  imgs = gr.ImageEditor(sources='upload', type="pil", label='Human. Mask with pen or use auto-masking', interactive=True)