Update app.py

app.py

@@ -64,10 +64,10 @@ def generate_image(category, img1, img2, img3, height, width, img_guidance_scale
     wears = predict_clothing(input_images[1:])
     if len(wears)==1:
         dress = wears[0]
-        text = f"""<img><|image_1|></img> Replace the clothes with {dress} in <img><|image_2|></img>."""
+        text = f"""<img><|image_1|></img> Replace the {category}'s clothes with {dress} in <img><|image_2|></img>."""
     elif len(wears)==2:
         topwear, bottomwear = wears[0], wears[1]
-        text = f"""<img><|image_1|></img> Replace the topwear with {topwear} in <img><|image_2|></img> and the bottomwear with {bottomwear} in <img><|image_3|></img>."""
+        text = f"""<img><|image_1|></img> Replace the {category}'s top with {topwear} in <img><|image_2|></img> and the {category}'s bottoms with {bottomwear} in <img><|image_3|></img>."""
     else:
         input_images = None