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