Satyajithchary commited on
Commit
010415d
1 Parent(s): 0b09501

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -4,9 +4,7 @@ from sam2.sam2_image_predictor import SAM2ImagePredictor
4
 
5
  predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large")
6
 
7
- with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
8
- predictor.set_image(<your_image>)
9
- masks, _, _ = predictor.predict(<input_prompts>)
10
  import gradio as gr
11
  import torch
12
  import numpy as np
 
4
 
5
  predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large")
6
 
7
+
 
 
8
  import gradio as gr
9
  import torch
10
  import numpy as np