mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-30 20:24:31 +00:00
fix: export directory issue
This commit is contained in:
@@ -4,6 +4,8 @@
|
||||
""" Utility functions for creating estimator. """
|
||||
|
||||
from pathlib import Path
|
||||
from os.path import join
|
||||
from tempfile import gettempdir
|
||||
|
||||
# pylint: disable=import-error
|
||||
import tensorflow as tf
|
||||
@@ -15,7 +17,7 @@ from ..model import model_fn
|
||||
from ..model.provider import get_default_model_provider
|
||||
|
||||
# Default exporting directory for predictor.
|
||||
DEFAULT_EXPORT_DIRECTORY = '/tmp/serving'
|
||||
DEFAULT_EXPORT_DIRECTORY = join(gettempdir(), 'serving')
|
||||
|
||||
|
||||
def create_estimator(params, MWF):
|
||||
|
||||
Reference in New Issue
Block a user