Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -64,10 +64,10 @@ def generate_image(category, img1, img2, img3, height, width, img_guidance_scale
|
|
64 |
wears = predict_clothing(input_images[1:])
|
65 |
if len(wears)==1:
|
66 |
dress = wears[0]
|
67 |
-
text = f"""<img><|image_1|></img>
|
68 |
elif len(wears)==2:
|
69 |
topwear, bottomwear = wears[0], wears[1]
|
70 |
-
text = f"""<img><|image_1|></img>
|
71 |
else:
|
72 |
input_images = None
|
73 |
|
|
|
64 |
wears = predict_clothing(input_images[1:])
|
65 |
if len(wears)==1:
|
66 |
dress = wears[0]
|
67 |
+
text = f"""The {category} in <img><|image_1|></img> standing, wearing {dress} in <img><|image_2|></img>, with a completely white background."""
|
68 |
elif len(wears)==2:
|
69 |
topwear, bottomwear = wears[0], wears[1]
|
70 |
+
text = f"""The {category} in <img><|image_1|></img> standing, wearing {topwear} in <img><|image_2|></img> and {bottomwear} in <img><|image_3|></img>, , with a completely white background."""
|
71 |
else:
|
72 |
input_images = None
|
73 |
|