Spaces:
Runtime error
Runtime error
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"""
|
68 |
elif len(wears)==2:
|
69 |
topwear, bottomwear = wears[0], wears[1]
|
70 |
-
text = f"""{topwear} in <img><|image_2|></img> and {bottomwear} in <img><|image_3|></img> on the {category} in <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"""Put the {dress} in <img><|image_2|></img> on the {category} in <img><|image_1|></img>"""
|
68 |
elif len(wears)==2:
|
69 |
topwear, bottomwear = wears[0], wears[1]
|
70 |
+
text = f"""Put the {topwear} in <img><|image_2|></img> and the {bottomwear} in <img><|image_3|></img> on the {category} in <img><|image_1|></img>"""
|
71 |
else:
|
72 |
input_images = None
|
73 |
|