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