Update app.py

app.py

@@ -62,10 +62,10 @@ def generate_image(category, img1, img2, img3, height, width, img_guidance_scale
         input_images = None
 
     #wears = predict_clothing(input_images[1:])
-    if len(wears)==1:
+    if len(input_images)==2:
         #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:
+    elif len(input_images)==3:
         #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: