generated from thinkode/modelRepository
not removing output folder when interrupting
This commit is contained in:
@@ -1,9 +1,7 @@
|
||||
import demucs.separate
|
||||
import shutil
|
||||
import argparse
|
||||
import os
|
||||
import shutil
|
||||
import json
|
||||
from log import log_step
|
||||
from genericpath import exists
|
||||
|
||||
@@ -19,11 +17,11 @@ try:
|
||||
# Options analyzing
|
||||
args = parser.parse_args()
|
||||
|
||||
log_step('init', 100, json.dumps({
|
||||
log_step('init', 100, {
|
||||
"audioFile": args.audio,
|
||||
"outputFolder": args.folder,
|
||||
"model": args.model,
|
||||
}))
|
||||
})
|
||||
|
||||
# Separate vocals with progress
|
||||
demucs.separate.main([
|
||||
@@ -37,8 +35,6 @@ try:
|
||||
|
||||
except KeyboardInterrupt:
|
||||
# Delete the output folder
|
||||
if exists(args.folder):
|
||||
shutil.rmtree(args.folder)
|
||||
log_step("exit", 100, "program exit")
|
||||
|
||||
except Exception as e:
|
||||
|
||||
Reference in New Issue
Block a user