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

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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(raw_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")
 
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")