Spaces:
Build error
Build error
Fix return type error
Browse files
app.py
CHANGED
|
@@ -361,10 +361,15 @@ def start_tryon(dict,garm_img,garment_des,is_checked,denoise_steps,seed, is_chec
|
|
| 361 |
|
| 362 |
|
| 363 |
# DensePose ๋ชจ๋ธ ๋ค์ด๋ก๋ ๋ฐ ๊ฒฝ๋ก ์ค์
|
| 364 |
-
densepose_model_path =
|
| 365 |
-
|
| 366 |
-
|
| 367 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 368 |
|
| 369 |
args = apply_net.create_argument_parser().parse_args(('show', './configs/densepose_rcnn_R_50_FPN_s1x.yaml', densepose_model_path, 'dp_segm', '-v', '--opts', 'MODEL.DEVICE', 'cuda'))
|
| 370 |
# verbosity = getattr(args, "verbosity", None)
|
|
|
|
| 361 |
|
| 362 |
|
| 363 |
# DensePose ๋ชจ๋ธ ๋ค์ด๋ก๋ ๋ฐ ๊ฒฝ๋ก ์ค์
|
| 364 |
+
densepose_model_path = './ckpt/densepose/model_final_162be9.pkl'
|
| 365 |
+
|
| 366 |
+
# ๋ชจ๋ธ ํ์ผ์ด ์์ผ๋ฉด ๋ค์ด๋ก๋ ์๋
|
| 367 |
+
if not os.path.exists(densepose_model_path):
|
| 368 |
+
print("DensePose model not found, attempting to download...")
|
| 369 |
+
download_success = download_densepose_model()
|
| 370 |
+
if not download_success:
|
| 371 |
+
print("Failed to download DensePose model")
|
| 372 |
+
return None, None
|
| 373 |
|
| 374 |
args = apply_net.create_argument_parser().parse_args(('show', './configs/densepose_rcnn_R_50_FPN_s1x.yaml', densepose_model_path, 'dp_segm', '-v', '--opts', 'MODEL.DEVICE', 'cuda'))
|
| 375 |
# verbosity = getattr(args, "verbosity", None)
|