Spaces:
Sleeping
Sleeping
Commit
·
8e6bca2
1
Parent(s):
a907317
app.py: enlarge target size for small reference image.
Browse files
app.py
CHANGED
|
@@ -256,7 +256,7 @@ def generate_image (
|
|
| 256 |
seed,
|
| 257 |
scheduler,
|
| 258 |
#enable_LCM,
|
| 259 |
-
enhance_face_region,
|
| 260 |
progress=gr.Progress(track_tqdm=True),
|
| 261 |
):
|
| 262 |
if face_image_path is None:
|
|
@@ -280,7 +280,13 @@ def generate_image (
|
|
| 280 |
|
| 281 |
ref_image_path = pose_image_path if pose_image_path else face_image_path
|
| 282 |
ref_image = PIL.Image.open(ref_image_path)
|
|
|
|
| 283 |
width, height = ref_image.size
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 284 |
|
| 285 |
CONTROLNET_DICT['pose']['strength'] = pose_strength
|
| 286 |
CONTROLNET_DICT['canny']['strength'] = canny_strength
|
|
@@ -292,7 +298,7 @@ def generate_image (
|
|
| 292 |
},
|
| 293 |
'model_name': f'{model_name}.safetensors',
|
| 294 |
'face_image_assets_ids': client.upload_assets([face_image_path]),
|
| 295 |
-
'ref_image_assets_ids': client.upload_assets([
|
| 296 |
'prompt': prompt,
|
| 297 |
'negative_prompt': negative_prompt,
|
| 298 |
'controlnet': {
|
|
@@ -493,10 +499,10 @@ with gr.Blocks(css=css) as demo:
|
|
| 493 |
scheduler = gr.Dropdown(
|
| 494 |
label='Schedulers',
|
| 495 |
choices=schedulers,
|
| 496 |
-
value='Euler
|
| 497 |
)
|
| 498 |
randomize_seed = gr.Checkbox(label='Randomize seed', value=True)
|
| 499 |
-
enhance_face_region = gr.Checkbox(label='Enhance non-face region', value=True)
|
| 500 |
|
| 501 |
with gr.Column(scale=1):
|
| 502 |
gallery = gr.Image(label='Generated Images')
|
|
@@ -534,7 +540,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 534 |
seed,
|
| 535 |
scheduler,
|
| 536 |
#enable_LCM,
|
| 537 |
-
enhance_face_region,
|
| 538 |
],
|
| 539 |
outputs=[gallery, usage_tips],
|
| 540 |
)
|
|
|
|
| 256 |
seed,
|
| 257 |
scheduler,
|
| 258 |
#enable_LCM,
|
| 259 |
+
#enhance_face_region,
|
| 260 |
progress=gr.Progress(track_tqdm=True),
|
| 261 |
):
|
| 262 |
if face_image_path is None:
|
|
|
|
| 280 |
|
| 281 |
ref_image_path = pose_image_path if pose_image_path else face_image_path
|
| 282 |
ref_image = PIL.Image.open(ref_image_path)
|
| 283 |
+
|
| 284 |
width, height = ref_image.size
|
| 285 |
+
large_edge = max(width, height)
|
| 286 |
+
if large_edge < 1024:
|
| 287 |
+
scaling = 1024 / large_edge
|
| 288 |
+
width = int(width * scaling)
|
| 289 |
+
height = int(height * scaling)
|
| 290 |
|
| 291 |
CONTROLNET_DICT['pose']['strength'] = pose_strength
|
| 292 |
CONTROLNET_DICT['canny']['strength'] = canny_strength
|
|
|
|
| 298 |
},
|
| 299 |
'model_name': f'{model_name}.safetensors',
|
| 300 |
'face_image_assets_ids': client.upload_assets([face_image_path]),
|
| 301 |
+
'ref_image_assets_ids': client.upload_assets([ref_image_path]),
|
| 302 |
'prompt': prompt,
|
| 303 |
'negative_prompt': negative_prompt,
|
| 304 |
'controlnet': {
|
|
|
|
| 499 |
scheduler = gr.Dropdown(
|
| 500 |
label='Schedulers',
|
| 501 |
choices=schedulers,
|
| 502 |
+
value='Euler',
|
| 503 |
)
|
| 504 |
randomize_seed = gr.Checkbox(label='Randomize seed', value=True)
|
| 505 |
+
#enhance_face_region = gr.Checkbox(label='Enhance non-face region', value=True)
|
| 506 |
|
| 507 |
with gr.Column(scale=1):
|
| 508 |
gallery = gr.Image(label='Generated Images')
|
|
|
|
| 540 |
seed,
|
| 541 |
scheduler,
|
| 542 |
#enable_LCM,
|
| 543 |
+
#enhance_face_region,
|
| 544 |
],
|
| 545 |
outputs=[gallery, usage_tips],
|
| 546 |
)
|