Spaces:
Runtime error
Runtime error
flamehaze1115
commited on
Update gradio_app.py
Browse files- 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()
|