flamehaze1115 commited on
Commit
a725b60
·
verified ·
1 Parent(s): 2842be9

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -62,7 +62,7 @@ def sam_init():
62
  return model, processor
63
 
64
  def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):
65
- bbox = np.array(bbox_coords)
66
  image = np.asarray(input_image)
67
 
68
  start_time = time.time()
 
62
  return model, processor
63
 
64
  def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):
65
+ bbox = [bbox_coords]
66
  image = np.asarray(input_image)
67
 
68
  start_time = time.time()