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 |
|