Spaces:
Runtime error
Runtime error
fix quality and make options more specific
Browse files
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 |
-
"
|
15 |
-
"
|
|
|
|
|
16 |
"Summer dress",
|
17 |
-
"
|
|
|
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 |
)
|