Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
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 |
|