fffiloni commited on
Commit
f82120a
·
verified ·
1 Parent(s): e61c4ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.value
264
- frame_names = stored_frame_names.value
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)