NikhilJoson commited on
Commit
7466a03
·
verified ·
1 Parent(s): d967414

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,10 +57,10 @@ def generate_image(category, img1, img2, img3, height, width, img_guidance_scale
57
  wears = predict_clothing(input_images[1:])
58
  if len(wears)==1:
59
  dress = wears[0]
60
- text = f"""The {category} in <img><|image_1|></img> is standing, wearing a {dress}. The {dress} is in <img><|image_2|></img>."""
61
  elif len(wears)==2:
62
  topwear, bottomwear = wears[0], wears[1]
63
- text = f"""The {category} in <img><|image_1|></img> is standing, wearing a {topwear} and a {bottomwear}.
64
  The {topwear} is in <img><|image_2|></img>. The {bottomwear} is in <img><|image_3|></img>."""
65
  else:
66
  input_images = None
 
57
  wears = predict_clothing(input_images[1:])
58
  if len(wears)==1:
59
  dress = wears[0]
60
+ text = f"""A {category} in a {dress}. The {category} is in <img><|image_1|></img>. The {dress} is in <img><|image_2|></img>."""
61
  elif len(wears)==2:
62
  topwear, bottomwear = wears[0], wears[1]
63
+ text = f"""A {category} in a {topwear} and a {bottomwear}. The {category} is in <img><|image_1|></img>.
64
  The {topwear} is in <img><|image_2|></img>. The {bottomwear} is in <img><|image_3|></img>."""
65
  else:
66
  input_images = None