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 |
|