jhj0517 commited on
Commit
bf46b63
·
1 Parent(s): 6f6c89b

Apply path constants

Browse files
modules/translation/translation_base.py CHANGED
@@ -8,13 +8,13 @@ from datetime import datetime
8
  from modules.whisper.whisper_parameter import *
9
  from modules.utils.subtitle_manager import *
10
  from modules.utils.files_manager import load_yaml, save_yaml
11
- from modules.utils.paths import DEFAULT_PARAMETERS_CONFIG_PATH
12
 
13
 
14
  class TranslationBase(ABC):
15
  def __init__(self,
16
- model_dir: str = os.path.join("models", "NLLB"),
17
- output_dir: str = os.path.join("outputs", "translations")
18
  ):
19
  super().__init__()
20
  self.model = None
 
8
  from modules.whisper.whisper_parameter import *
9
  from modules.utils.subtitle_manager import *
10
  from modules.utils.files_manager import load_yaml, save_yaml
11
+ from modules.utils.paths import DEFAULT_PARAMETERS_CONFIG_PATH, NLLB_MODELS_DIR, TRANSLATION_OUTPUT_DIR
12
 
13
 
14
  class TranslationBase(ABC):
15
  def __init__(self,
16
+ model_dir: str = NLLB_MODELS_DIR,
17
+ output_dir: str = TRANSLATION_OUTPUT_DIR
18
  ):
19
  super().__init__()
20
  self.model = None