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