nami0342 commited on
Commit
f264da1
ยท
1 Parent(s): 6614b04

Fix return type error

Browse files
Files changed (1) hide show
  1. app.py +9 -4
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 = download_densepose_model()
365
- if densepose_model_path is None:
366
- print("Failed to download DensePose model, using default path")
367
- densepose_model_path = './ckpt/densepose/model_final_162be9.pkl'
 
 
 
 
 
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)