Spaces:
Sleeping
Sleeping
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)
|