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