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
|