Spaces:
Running
Running
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 =
|
17 |
-
output_dir: str =
|
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
|