NikhilJoson commited on
Commit
e40e56f
·
verified ·
1 Parent(s): a854c9a

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(img1, img2, img3, height, width, img_guidance_scale, inferenc
57
  wears = predict_clothing(input_images[1:])
58
  if len(wears)==1:
59
  dress = wears[0]
60
- text = """A male wearing a {dress}. The male 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 = """A male wearing a {topwear} and a {bottomwear}. The male 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
 
57
  wears = predict_clothing(input_images[1:])
58
  if len(wears)==1:
59
  dress = wears[0]
60
+ text = f"""A male wearing a {dress}. The male 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 male wearing a {topwear} and a {bottomwear}. The male 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