NikhilJoson commited on
Commit
9ce2c5c
Β·
verified Β·
1 Parent(s): 5c8060f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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"""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"""{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