Update app.py

app.py

@@ -61,13 +61,13 @@ def generate_image(category, img1, img2, img3, height, width, img_guidance_scale
     if len(input_images) == 0:
         input_images = None
 
-    wears = predict_clothing(input_images[1:])
+    #wears = predict_clothing(input_images[1:])
     if len(wears)==1:
-        dress = wears[0]
-        text = f"""The {category} in <img><|image_1|></img> standing, wearing {dress} in <img><|image_2|></img>, with a completely white background."""
+        #dress = wears[0]
+        text = f"""A {category} wearing a dress, with completely white background. The {category} is in <img><|image_1|></img>. The dress is in <img><|image_2|></img>."""
     elif len(wears)==2:
-        topwear, bottomwear = wears[0], wears[1]
-        text = f"""Take the {category} from <img><|image_1|></img>, and dress them with the topwear in <img><|image_2|></img> and the bottomwear in <img><|image_3|></img>."""
+        #topwear, bottomwear = wears[0], wears[1]
+        text = f"""A {category} wearing a Black tshirt and Olive pants, with completely white background. The {category} is in <img><|image_1|></img>. The Black tshirt is in <img><|image_2|></img>. The Olive pants is in <img><|image_3|></img>."""
     else:
         input_images = None