mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-31 14:58:23 +00:00
🐛 fix path, entrypoint and logging
This commit is contained in:
@@ -112,8 +112,8 @@ def separate(
|
|||||||
stft_backend=stft_backend)
|
stft_backend=stft_backend)
|
||||||
for filename in files:
|
for filename in files:
|
||||||
separator.separate_to_file(
|
separator.separate_to_file(
|
||||||
filename,
|
str(filename),
|
||||||
output_path,
|
str(output_path),
|
||||||
audio_adapter=audio_adapter,
|
audio_adapter=audio_adapter,
|
||||||
offset=offset,
|
offset=offset,
|
||||||
duration=duration,
|
duration=duration,
|
||||||
@@ -221,8 +221,13 @@ def evaluate(
|
|||||||
return metrics
|
return metrics
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
def entrypoint():
|
||||||
|
""" Application entrypoint. """
|
||||||
try:
|
try:
|
||||||
spleeter()
|
spleeter()
|
||||||
except SpleeterError as e:
|
except SpleeterError as e:
|
||||||
logger.error(e)
|
logger.error(e)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
entrypoint()
|
||||||
|
|||||||
@@ -10,9 +10,6 @@ from os import environ
|
|||||||
|
|
||||||
# pyright: reportMissingImports=false
|
# pyright: reportMissingImports=false
|
||||||
# pylint: disable=import-error
|
# pylint: disable=import-error
|
||||||
import tensorflow as tf
|
|
||||||
|
|
||||||
from tensorflow.compat.v1 import logging as tf_logging
|
|
||||||
from typer import echo
|
from typer import echo
|
||||||
# pylint: enable=import-error
|
# pylint: enable=import-error
|
||||||
|
|
||||||
@@ -46,7 +43,9 @@ def configure_logger(verbose: bool) -> None:
|
|||||||
verbose (bool):
|
verbose (bool):
|
||||||
`True` to use verbose logger, `False` otherwise.
|
`True` to use verbose logger, `False` otherwise.
|
||||||
"""
|
"""
|
||||||
tf_logger = tf.get_logger()
|
from tensorflow import get_logger
|
||||||
|
from tensorflow.compat.v1 import logging as tf_logging
|
||||||
|
tf_logger = get_logger()
|
||||||
tf_logger.handlers = [handler]
|
tf_logger.handlers = [handler]
|
||||||
if verbose:
|
if verbose:
|
||||||
tf_logging.set_verbosity(tf_logging.INFO)
|
tf_logging.set_verbosity(tf_logging.INFO)
|
||||||
|
|||||||
Reference in New Issue
Block a user