NikhilJoson commited on
Commit
7c5b028
ยท
verified ยท
1 Parent(s): a2798fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -203,7 +203,7 @@ def main_(imgs,topwear_img,topwear_des,bottomwear_img,bottomwear_des,dress_img,d
203
  return start_tryon(imgs,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
204
  elif topwear_img!=None and bottomwear_img!=None:
205
  _, half_img, half_mask = start_tryon(imgs,topwear_img,topwear_des,"upper_body",is_checked,is_checked_crop,denoise_steps,seed)
206
- return start_tryon(imgs,half_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
207
 
208
 
209
  garm_list = os.listdir(os.path.join(example_path,"cloth"))
@@ -238,19 +238,19 @@ with image_blocks as demo:
238
 
239
  with gr.Column():
240
  topwear_image = gr.Image(label="Topwear", sources='upload', type="pil")
241
- with gr.Row(elem_id="prompt-container"):
242
  with gr.Row():
243
  topwear_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
244
  example = gr.Examples(inputs=topwear_image, examples_per_page=8,examples=garm_list_path)
245
  with gr.Column():
246
  bottomwear_image = gr.Image(label="Bottomwear", sources='upload', type="pil")
247
- with gr.Row(elem_id="prompt-container"):
248
  with gr.Row():
249
  bottomwear_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
250
  example = gr.Examples(inputs=bottomwear_image, examples_per_page=8, examples=garm_list_path)
251
  with gr.Column():
252
  dress_image = gr.Image(label="Dress", sources='upload', type="pil")
253
- with gr.Row(elem_id="prompt-container"):
254
  with gr.Row():
255
  dress_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
256
  example = gr.Examples(inputs=dress_image, examples_per_page=8, examples=garm_list_path)
 
203
  return start_tryon(imgs,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
204
  elif topwear_img!=None and bottomwear_img!=None:
205
  _, half_img, half_mask = start_tryon(imgs,topwear_img,topwear_des,"upper_body",is_checked,is_checked_crop,denoise_steps,seed)
206
+ return start_tryon(half_img,bottomwear_img,bottomwear_des,"lower_body",is_checked,is_checked_crop,denoise_steps,seed)
207
 
208
 
209
  garm_list = os.listdir(os.path.join(example_path,"cloth"))
 
238
 
239
  with gr.Column():
240
  topwear_image = gr.Image(label="Topwear", sources='upload', type="pil")
241
+ with gr.Row(elem_id="topwear-prompt-container"):
242
  with gr.Row():
243
  topwear_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
244
  example = gr.Examples(inputs=topwear_image, examples_per_page=8,examples=garm_list_path)
245
  with gr.Column():
246
  bottomwear_image = gr.Image(label="Bottomwear", sources='upload', type="pil")
247
+ with gr.Row(elem_id="bottomwear-prompt-container"):
248
  with gr.Row():
249
  bottomwear_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
250
  example = gr.Examples(inputs=bottomwear_image, examples_per_page=8, examples=garm_list_path)
251
  with gr.Column():
252
  dress_image = gr.Image(label="Dress", sources='upload', type="pil")
253
+ with gr.Row(elem_id="dress-prompt-container"):
254
  with gr.Row():
255
  dress_desc = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
256
  example = gr.Examples(inputs=dress_image, examples_per_page=8, examples=garm_list_path)