Update gradio_app.py

gradio_app.py

@@ -62,7 +62,8 @@ def sam_init():
     return model, processor
 
 def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):
-    bbox = [bbox_coords]
+    bbox = torch.tensor(bbox_coords, dtype=torch.float32)
+    bbox = box_tensor = box_tensor.unsqueeze(0).unsqueeze(0) # (1, 1, 4)
     image = np.asarray(input_image)
 
     start_time = time.time()