flamehaze1115 commited on
Commit
28468ea
·
verified ·
1 Parent(s): 080cd4b

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -63,7 +63,7 @@ def sam_init():
63
 
64
  def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):
65
  bbox = torch.tensor(bbox_coords, dtype=torch.float32)
66
- bbox = bbox.unsqueeze(0)
67
  image = np.asarray(input_image)
68
 
69
  start_time = time.time()
 
63
 
64
  def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):
65
  bbox = torch.tensor(bbox_coords, dtype=torch.float32)
66
+ bbox = bbox.unsqueeze(0).unsqueeze(0)
67
  image = np.asarray(input_image)
68
 
69
  start_time = time.time()