Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -260,8 +260,8 @@ def propagate_to_all(checkpoint, stored_inference_state, stored_frame_names):
|
|
260 |
#### PROPAGATION ####
|
261 |
sam2_checkpoint, model_cfg = load_model(checkpoint)
|
262 |
predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
|
263 |
-
inference_state = stored_inference_state
|
264 |
-
frame_names = stored_frame_names
|
265 |
# Define a directory to save the JPEG images
|
266 |
frames_output_dir = "frames_output_images"
|
267 |
os.makedirs(frames_output_dir, exist_ok=True)
|
|
|
260 |
#### PROPAGATION ####
|
261 |
sam2_checkpoint, model_cfg = load_model(checkpoint)
|
262 |
predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
|
263 |
+
inference_state = stored_inference_state
|
264 |
+
frame_names = stored_frame_names
|
265 |
# Define a directory to save the JPEG images
|
266 |
frames_output_dir = "frames_output_images"
|
267 |
os.makedirs(frames_output_dir, exist_ok=True)
|