Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -275,7 +275,7 @@ def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_
|
|
275 |
|
276 |
return "output_first_frame.jpg", frame_names, inference_state
|
277 |
|
278 |
-
def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type):
|
279 |
#### PROPAGATION ####
|
280 |
sam2_checkpoint, model_cfg = load_model(checkpoint)
|
281 |
predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
|
|
|
275 |
|
276 |
return "output_first_frame.jpg", frame_names, inference_state
|
277 |
|
278 |
+
def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type, progress=gr.Progress(track_tqdm=True)):
|
279 |
#### PROPAGATION ####
|
280 |
sam2_checkpoint, model_cfg = load_model(checkpoint)
|
281 |
predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
|