fffiloni commited on
Commit
97539a7
·
verified ·
1 Parent(s): f065d89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -379,7 +379,7 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
379
  show_white_mask(out_mask, ax)
380
 
381
  # Save the white mask figure to an image in memory
382
- mask_filename = os.path.join(masks_output_dir, f"mask_{out_frame_idx}.jpg")
383
  fig.savefig(mask_filename, format='jpg', bbox_inches="tight", pad_inches=0)
384
  plt.close(fig)
385
 
 
379
  show_white_mask(out_mask, ax)
380
 
381
  # Save the white mask figure to an image in memory
382
+ mask_filename = os.path.join(mask_frames_output_dir, f"mask_{out_frame_idx}.jpg")
383
  fig.savefig(mask_filename, format='jpg', bbox_inches="tight", pad_inches=0)
384
  plt.close(fig)
385