diff --git a/spleeter/utils/estimator.py b/spleeter/utils/estimator.py index a908886..95c4219 100644 --- a/spleeter/utils/estimator.py +++ b/spleeter/utils/estimator.py @@ -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):