Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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", "
|
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)
|