dikarel commited on
Commit
6a40b6d
·
1 Parent(s): 1a9bd9f

fix quality and make options more specific

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -7,14 +7,14 @@ from PIL.Image import Image as PILImage
7
  from lib.resize_image import resize_image
8
 
9
  OUTFIT_SELECTION = [
10
- "Athletic wear",
11
- "Ball gown",
12
- "Bridal gown",
13
  "Camel trench coat",
14
- "Formal suit",
15
- "Haute couture",
 
 
16
  "Summer dress",
17
- "Winter coat",
 
18
  ]
19
 
20
 
@@ -46,7 +46,7 @@ def generate_output(img_input: PILImage, drp_outfit: str) -> PILImage:
46
 
47
  people_mask = find_people(img_input)
48
  img_output = redraw_image(
49
- prompt=f"person wearing {drp_outfit}",
50
  image=img_input,
51
  mask=people_mask,
52
  )
 
7
  from lib.resize_image import resize_image
8
 
9
  OUTFIT_SELECTION = [
 
 
 
10
  "Camel trench coat",
11
+ "Fuzzy white winter coat",
12
+ "Golden ball gown with tiara",
13
+ "Mondrian-inspired haute couture",
14
+ "Rock and roll t-shirt",
15
  "Summer dress",
16
+ "Tie dye t-shirt",
17
+ "Velvet dinner jacket",
18
  ]
19
 
20
 
 
46
 
47
  people_mask = find_people(img_input)
48
  img_output = redraw_image(
49
+ prompt=f"best quality. high 4k resolution. person wearing {drp_outfit}",
50
  image=img_input,
51
  mask=people_mask,
52
  )