NikhilJoson commited on
Commit
9d0e65d
·
verified ·
1 Parent(s): 15d8b55

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"""<img><|image_1|></img> Replace the {category}'s clothes with {dress} in <img><|image_2|></img>."""
68
  elif len(wears)==2:
69
  topwear, bottomwear = wears[0], wears[1]
70
- text = f"""<img><|image_1|></img> Replace the {category}'s top-wear with {topwear} in <img><|image_2|></img> and the {category}'s bottom-wear with {bottomwear} in <img><|image_3|></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"""The {category} in <img><|image_1|></img> standing, wearing {dress} in <img><|image_2|></img>, with a completely white background."""
68
  elif len(wears)==2:
69
  topwear, bottomwear = wears[0], wears[1]
70
+ text = f"""The {category} in <img><|image_1|></img> standing, wearing {topwear} in <img><|image_2|></img> and {bottomwear} in <img><|image_3|></img>, , with a completely white background."""
71
  else:
72
  input_images = None
73