🎨 ⬆️ add fix dep and clean code

setup.py

@@ -14,7 +14,7 @@ __license__ = 'MIT License'
 
 # Default project values.
 project_name = 'spleeter'
-project_version = '2.0.1'
+project_version = '2.0.2'
 tensorflow_dependency = 'tensorflow'
 tensorflow_version = '2.3.0'
 here = path.abspath(path.dirname(__file__))
@@ -50,11 +50,13 @@ setup(
     python_requires='>=3.6, <3.9',
     include_package_data=True,
     install_requires=[
-        'ffmpeg-python',
+        'ffmpeg-python==0.2.0',
         'importlib_resources ; python_version<"3.7"',
         'norbert==0.2.1',
+        'numpy<1.19.0,>=1.16.0',
         'pandas==1.1.2',
         'requests',
+        'scipy==1.4.1',
         'setuptools>=41.0.0',
         'librosa==0.8.0',
         '{}=={}'.format(tensorflow_dependency, tensorflow_version),

spleeter/utils/estimator.py

@@ -3,18 +3,12 @@
 
 """ Utility functions for creating estimator. """
 
-from pathlib import Path
-from os.path import join
-
-# pylint: disable=import-error
-import tensorflow as tf
-
+import tensorflow as tf  # pylint: disable=import-error
 
 from ..model import model_fn
 from ..model.provider import get_default_model_provider
 
 
-
 def get_default_model_dir(model_dir):
     """
     Transforms a string like 'spleeter:2stems' into an actual path.
@@ -24,6 +18,7 @@ def get_default_model_dir(model_dir):
     model_provider = get_default_model_provider()
     return model_provider.get(model_dir)
 
+
 def create_estimator(params, MWF):
     """
         Initialize tensorflow estimator that will perform separation
@@ -35,8 +30,6 @@ def create_estimator(params, MWF):
             a tensorflow estimator
     """
     # Load model.
-
-
     params['model_dir'] = get_default_model_dir(params['model_dir'])
     params['MWF'] = MWF
     # Setup config