Spaces:
Runtime error
Runtime error
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:
|