dryade36513 commited on
Commit
07b5520
·
verified ·
1 Parent(s): b5c82de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -14
app.py CHANGED
@@ -175,8 +175,8 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
175
  # Extract the images
176
  with torch.cuda.amp.autocast():
177
  with torch.no_grad():
178
- prompt = "model is wearing " + garment_des
179
- negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality"
180
  with torch.inference_mode():
181
  (
182
  prompt_embeds,
@@ -190,8 +190,8 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
190
  negative_prompt=negative_prompt,
191
  )
192
 
193
- prompt = "a photo of " + garment_des
194
- negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality"
195
  if not isinstance(prompt, List):
196
  prompt = [prompt] * 1
197
  if not isinstance(negative_prompt, List):
@@ -264,26 +264,26 @@ with image_blocks as demo:
264
  gr.Markdown("使用您的影像和服裝影像進行虛擬試穿")
265
  with gr.Row():
266
  with gr.Column():
267
- imgs = gr.ImageEditor(sources='upload', type="pil", label='使用畫筆或自動遮罩替模特兒生成遮罩', interactive=True)
268
  with gr.Row():
269
- is_checked = gr.Checkbox(label="Yes", info="自動生成遮罩 (僅需五秒)",value=True)
270
  with gr.Row():
271
- is_checked_crop = gr.Checkbox(label="Yes", info="自動裁切",value=False)
272
 
273
  example = gr.Examples(
274
  inputs=imgs,
275
- examples_per_page=10,
276
  examples=human_ex_list
277
  )
278
 
279
  with gr.Column():
280
- garm_img = gr.Image(label="Garment", sources='upload', type="pil")
281
  with gr.Row(elem_id="prompt-container"):
282
  with gr.Row():
283
- prompt = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
284
  example = gr.Examples(
285
  inputs=garm_img,
286
- examples_per_page=10,
287
  examples=garm_list_path)
288
  with gr.Column():
289
  # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
@@ -296,11 +296,11 @@ with image_blocks as demo:
296
 
297
 
298
  with gr.Column():
299
- try_button = gr.Button(value="Try-on")
300
- with gr.Accordion(label="Advanced Settings", open=False):
301
  with gr.Row():
302
  denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=40, value=30, step=1)
303
- seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=42)
304
 
305
 
306
 
 
175
  # Extract the images
176
  with torch.cuda.amp.autocast():
177
  with torch.no_grad():
178
+ prompt = "((best quality, masterpiece, ultra-detailed, high quality photography, photo realistic)), the model is wearing " + garment_des
179
+ negative_prompt = "monochrome, lowres, bad anatomy, worst quality, normal quality, low quality, blurry, jpeg artifacts, sketch"
180
  with torch.inference_mode():
181
  (
182
  prompt_embeds,
 
190
  negative_prompt=negative_prompt,
191
  )
192
 
193
+ prompt = "((best quality, masterpiece, ultra-detailed, high quality photography, photo realistic)), a photo of " + garment_des
194
+ negative_prompt = "monochrome, lowres, bad anatomy, worst quality, normal quality, low quality, blurry, jpeg artifacts, sketch"
195
  if not isinstance(prompt, List):
196
  prompt = [prompt] * 1
197
  if not isinstance(negative_prompt, List):
 
264
  gr.Markdown("使用您的影像和服裝影像進行虛擬試穿")
265
  with gr.Row():
266
  with gr.Column():
267
+ imgs = gr.ImageEditor(sources='upload', type="pil", label='AI虛擬模特兒-請啟用全自動偵測模式或是使用畫筆在需要更衣的部位塗抹註記', interactive=True)
268
  with gr.Row():
269
+ is_checked = gr.Checkbox(label="Yes", info="啟用全自動偵測更衣模式",value=True)
270
  with gr.Row():
271
+ is_checked_crop = gr.Checkbox(label="Yes", info="開啟自動剪裁並調整圖片大小模式",value=False)
272
 
273
  example = gr.Examples(
274
  inputs=imgs,
275
+ examples_per_page=15,
276
  examples=human_ex_list
277
  )
278
 
279
  with gr.Column():
280
+ garm_img = gr.Image(label="成衣正鋪相片", sources='upload', type="pil")
281
  with gr.Row(elem_id="prompt-container"):
282
  with gr.Row():
283
+ prompt = gr.Textbox(label="(選填)輸入該衣物的英文描述", placeholder="Short Sleeve Round Neck T-shirts", show_label=True, elem_id="prompt")
284
  example = gr.Examples(
285
  inputs=garm_img,
286
+ examples_per_page=16,
287
  examples=garm_list_path)
288
  with gr.Column():
289
  # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
 
296
 
297
 
298
  with gr.Column():
299
+ try_button = gr.Button(value="更衣")
300
+ with gr.Accordion(label="進階設定", open=False):
301
  with gr.Row():
302
  denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=40, value=30, step=1)
303
+ seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=-1)
304
 
305
 
306