chaerinmin commited on
Commit
ac33442
·
1 Parent(s): f0d2205

hf cuda issue

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -665,6 +665,7 @@ def ready_sample(img_cropped, img_original, ex_mask, inpaint_mask, keypts, keypt
665
  img = cv2.resize(img_cropped["background"][..., :3], opts.image_size, interpolation=cv2.INTER_AREA)
666
  else:
667
  img = cv2.resize(img_original[..., :3], opts.image_size, interpolation=cv2.INTER_AREA)
 
668
  sam_predictor = init_sam(ckpt_path=sam_path, device="cuda")
669
  sam_predictor.set_image(img)
670
  if keypts is None and keypts_np is not None:
 
665
  img = cv2.resize(img_cropped["background"][..., :3], opts.image_size, interpolation=cv2.INTER_AREA)
666
  else:
667
  img = cv2.resize(img_original[..., :3], opts.image_size, interpolation=cv2.INTER_AREA)
668
+ gr.Info("Preparing... Please wait the process to be started.")
669
  sam_predictor = init_sam(ckpt_path=sam_path, device="cuda")
670
  sam_predictor.set_image(img)
671
  if keypts is None and keypts_np is not None: