Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
ac33442
1
Parent(s):
f0d2205
hf cuda issue
Browse files
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:
|