Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -90,9 +90,9 @@ hf_hub_download(
|
|
| 90 |
local_dir="/data/checkpoints",
|
| 91 |
)
|
| 92 |
# download antelopev2
|
| 93 |
-
|
| 94 |
-
|
| 95 |
-
os.system("unzip /data/antelopev2.zip -d /data/models/")
|
| 96 |
|
| 97 |
app = FaceAnalysis(name='antelopev2', root='/data', providers=['CPUExecutionProvider'])
|
| 98 |
app.prepare(ctx_id=0, det_size=(640, 640))
|
|
@@ -432,17 +432,18 @@ with gr.Blocks(css="custom.css") as demo:
|
|
| 432 |
queue=False,
|
| 433 |
show_progress=False
|
| 434 |
)
|
| 435 |
-
|
| 436 |
-
fn=check_selected,
|
| 437 |
-
inputs=[selected_state],
|
| 438 |
-
queue=False,
|
| 439 |
-
show_progress=False
|
| 440 |
-
).success(
|
| 441 |
fn=center_crop_image_as_square,
|
| 442 |
inputs=[photo],
|
| 443 |
outputs=[photo],
|
| 444 |
queue=False,
|
| 445 |
show_progress=False,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 446 |
).success(
|
| 447 |
fn=run_lora,
|
| 448 |
inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],
|
|
@@ -453,12 +454,6 @@ with gr.Blocks(css="custom.css") as demo:
|
|
| 453 |
inputs=[selected_state],
|
| 454 |
queue=False,
|
| 455 |
show_progress=False
|
| 456 |
-
).success(
|
| 457 |
-
fn=center_crop_image_as_square,
|
| 458 |
-
inputs=[photo],
|
| 459 |
-
outputs=[photo],
|
| 460 |
-
queue=False,
|
| 461 |
-
show_progress=False,
|
| 462 |
).success(
|
| 463 |
fn=run_lora,
|
| 464 |
inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],
|
|
|
|
| 90 |
local_dir="/data/checkpoints",
|
| 91 |
)
|
| 92 |
# download antelopev2
|
| 93 |
+
if not os.path.exists("/data/models/scrfd_10g_bnkps.onnx"):
|
| 94 |
+
gdown.download(url="https://drive.google.com/file/d/18wEUfMNohBJ4K3Ly5wpTejPfDzp-8fI8/view?usp=sharing", output="/data/", quiet=False, fuzzy=True)
|
| 95 |
+
os.system("unzip /data/antelopev2.zip -d /data/models/")
|
| 96 |
|
| 97 |
app = FaceAnalysis(name='antelopev2', root='/data', providers=['CPUExecutionProvider'])
|
| 98 |
app.prepare(ctx_id=0, det_size=(640, 640))
|
|
|
|
| 432 |
queue=False,
|
| 433 |
show_progress=False
|
| 434 |
)
|
| 435 |
+
photo.upload(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 436 |
fn=center_crop_image_as_square,
|
| 437 |
inputs=[photo],
|
| 438 |
outputs=[photo],
|
| 439 |
queue=False,
|
| 440 |
show_progress=False,
|
| 441 |
+
)
|
| 442 |
+
prompt.submit(
|
| 443 |
+
fn=check_selected,
|
| 444 |
+
inputs=[selected_state],
|
| 445 |
+
queue=False,
|
| 446 |
+
show_progress=False
|
| 447 |
).success(
|
| 448 |
fn=run_lora,
|
| 449 |
inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],
|
|
|
|
| 454 |
inputs=[selected_state],
|
| 455 |
queue=False,
|
| 456 |
show_progress=False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 457 |
).success(
|
| 458 |
fn=run_lora,
|
| 459 |
inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],
|