jhj0517 commited on
Commit
2d93272
·
1 Parent(s): c573a1a

add output_dir

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -19,8 +19,13 @@ class App:
19
  self.whisper_inf = self.init_whisper()
20
  print(f"Use \"{self.args.whisper_type}\" implementation")
21
  print(f"Device \"{self.whisper_inf.device}\" is detected")
22
- self.nllb_inf = NLLBInference()
23
- self.deepl_api = DeepLAPI()
 
 
 
 
 
24
 
25
  def init_whisper(self):
26
  whisper_type = self.args.whisper_type.lower().strip()
 
19
  self.whisper_inf = self.init_whisper()
20
  print(f"Use \"{self.args.whisper_type}\" implementation")
21
  print(f"Device \"{self.whisper_inf.device}\" is detected")
22
+ self.nllb_inf = NLLBInference(
23
+ model_dir=self.args.nllb_model_dir,
24
+ output_dir=self.args.output_dir
25
+ )
26
+ self.deepl_api = DeepLAPI(
27
+ output_dir=self.args.output_dir
28
+ )
29
 
30
  def init_whisper(self):
31
  whisper_type = self.args.whisper_type.lower().strip()