Hugging Face's logo

Spaces:
LPX55
/
Lightning-Painter-Multitool
Running on Zero

App Files Community
10
LPX55 commited on
Commit
6fbb74e
·
1 Parent(s): a232a77

hold

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ from PIL import Image, ImageDraw
11
  import numpy as np
12
  from sam2.sam2_image_predictor import SAM2ImagePredictor
13
 
14
- predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2.1-hiera-large")
15
- print(predictor)
16
  # with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
17
  # predictor.set_image(<your_image>)
18
  # masks, _, _ = predictor.predict(<input_prompts>)
 
11
  import numpy as np
12
  from sam2.sam2_image_predictor import SAM2ImagePredictor
13
 
14
+ # predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2.1-hiera-large", device="cuda")
15
+ # print(predictor)
16
  # with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
17
  # predictor.set_image(<your_image>)
18
  # masks, _, _ = predictor.predict(<input_prompts>)