Spaces:
Running
Running
jhj0517
commited on
Commit
·
aa11c47
1
Parent(s):
732a962
Add defaults to the function
Browse files
modules/translation/nllb_inference.py
CHANGED
|
@@ -35,7 +35,7 @@ class NLLBInference(TranslationBase):
|
|
| 35 |
model_size: str,
|
| 36 |
src_lang: str,
|
| 37 |
tgt_lang: str,
|
| 38 |
-
progress: gr.Progress
|
| 39 |
):
|
| 40 |
if model_size != self.current_model_size or self.model is None:
|
| 41 |
print("\nInitializing NLLB Model..\n")
|
|
|
|
| 35 |
model_size: str,
|
| 36 |
src_lang: str,
|
| 37 |
tgt_lang: str,
|
| 38 |
+
progress: gr.Progress = gr.Progress()
|
| 39 |
):
|
| 40 |
if model_size != self.current_model_size or self.model is None:
|
| 41 |
print("\nInitializing NLLB Model..\n")
|
modules/translation/translation_base.py
CHANGED
|
@@ -37,7 +37,7 @@ class TranslationBase(ABC):
|
|
| 37 |
model_size: str,
|
| 38 |
src_lang: str,
|
| 39 |
tgt_lang: str,
|
| 40 |
-
progress: gr.Progress
|
| 41 |
):
|
| 42 |
pass
|
| 43 |
|
|
|
|
| 37 |
model_size: str,
|
| 38 |
src_lang: str,
|
| 39 |
tgt_lang: str,
|
| 40 |
+
progress: gr.Progress = gr.Progress()
|
| 41 |
):
|
| 42 |
pass
|
| 43 |
|