Spaces:
Runtime error
Runtime error
flamehaze1115
commited on
Update gradio_app.py
Browse files- gradio_app.py +1 -1
gradio_app.py
CHANGED
@@ -67,7 +67,7 @@ def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):
|
|
67 |
|
68 |
start_time = time.time()
|
69 |
|
70 |
-
inputs = sam_processor(
|
71 |
outputs = sam_model(**inputs)
|
72 |
|
73 |
print(f"SAM Time: {time.time() - start_time:.3f}s")
|
|
|
67 |
|
68 |
start_time = time.time()
|
69 |
|
70 |
+
inputs = sam_processor(input_image, input_boxes=bbox, return_tensors="pt").to("cuda")
|
71 |
outputs = sam_model(**inputs)
|
72 |
|
73 |
print(f"SAM Time: {time.time() - start_time:.3f}s")
|