TiberiuCristianLeon commited on
Commit
aff7959
·
verified ·
1 Parent(s): 0d0d869

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ langs.extend(list(all_langs.keys())) # Language options as list, add favourite l
21
  iso1_to_name = {iso[1]: iso[0] for iso in non_empty_isos} # {'ro': 'Romanian', 'de': 'German'}
22
 
23
  def timer(func):
 
24
  def wrapper(*args, **kwargs):
25
- import time
26
  start_time = time.time()
27
  result = func(*args, **kwargs)
28
  end_time = time.time()
@@ -273,7 +273,7 @@ class Translators:
273
  model_path = snapshot_download(self.model_name)
274
 
275
  tokenizer = SentencePieceProcessor()
276
- tokenizer.load(f"{model_path}/sentencepiece.model")
277
  translator = ctranslate2.Translator(model_path)
278
 
279
  input_tokens = tokenizer.encode(f"<2{self.tl}> {self.input_text}", out_type=str)
 
21
  iso1_to_name = {iso[1]: iso[0] for iso in non_empty_isos} # {'ro': 'Romanian', 'de': 'German'}
22
 
23
  def timer(func):
24
+ import time
25
  def wrapper(*args, **kwargs):
 
26
  start_time = time.time()
27
  result = func(*args, **kwargs)
28
  end_time = time.time()
 
273
  model_path = snapshot_download(self.model_name)
274
 
275
  tokenizer = SentencePieceProcessor()
276
+ tokenizer.load(f"{model_path}/spiece.model")
277
  translator = ctranslate2.Translator(model_path)
278
 
279
  input_tokens = tokenizer.encode(f"<2{self.tl}> {self.input_text}", out_type=str)