yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: export directory issue

Browse Source
This commit is contained in:
Félix Voituret
2019-11-06 20:36:31 +01:00
parent 01319e1a63
commit 2559d52ecf
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
spleeter/utils/estimator.py
View File

@@ -4,6 +4,8 @@
""" Utility functions for creating estimator. """ """ Utility functions for creating estimator. """
from pathlib import Path from pathlib import Path
from os.path import join
from tempfile import gettempdir
# pylint: disable=import-error # pylint: disable=import-error
import tensorflow as tf import tensorflow as tf
@@ -15,7 +17,7 @@ from ..model import model_fn
from ..model.provider import get_default_model_provider from ..model.provider import get_default_model_provider
# Default exporting directory for predictor. # Default exporting directory for predictor.
DEFAULT_EXPORT_DIRECTORY = '/tmp/serving' DEFAULT_EXPORT_DIRECTORY = join(gettempdir(), 'serving')
def create_estimator(params, MWF): def create_estimator(params, MWF):