Gainward777 commited on
Commit
3718c2f
·
verified ·
1 Parent(s): f2dadba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -32,8 +32,7 @@ def generate_result(model_img:Image,
32
  images = sd_pipline(
33
  prompt=prompt,
34
  image=model_img,
35
- mask_image=cloth_mask,
36
- #ip_adapter_image=cloth_img,
37
  generator=generator,
38
  num_inference_steps=n_steps,
39
 
@@ -55,14 +54,13 @@ gr.Interface(
55
  title = 'Try-On Flux',
56
  inputs=[
57
  gr.Image(sources = 'upload', label='Model image', type = 'pil'),
58
- gr.Image(sources = 'upload', label='Mask', type = 'pil'),
59
- #gr.Textbox(label = 'Cloth class'),
60
  gr.Textbox(label = 'Prompt')
61
  ],
62
  outputs = [
63
  gr.Image()
64
  ],
65
  examples=[
66
- ["./examples/models/girl1.jpg","./examples/clothes/mask.png", "shirt", "black shirt"], #"./examples/clothes/t_short.jpg", "shirt", "black shirt"],
67
  ]
68
  ).launch(debug=True,share=True)
 
32
  images = sd_pipline(
33
  prompt=prompt,
34
  image=model_img,
35
+ mask_image=cloth_mask,
 
36
  generator=generator,
37
  num_inference_steps=n_steps,
38
 
 
54
  title = 'Try-On Flux',
55
  inputs=[
56
  gr.Image(sources = 'upload', label='Model image', type = 'pil'),
57
+ gr.Image(sources = 'upload', label='Mask', type = 'pil'),
 
58
  gr.Textbox(label = 'Prompt')
59
  ],
60
  outputs = [
61
  gr.Image()
62
  ],
63
  examples=[
64
+ ["./examples/models/girl1.jpg","./examples/clothes/mask.png", "black shirt"],
65
  ]
66
  ).launch(debug=True,share=True)