Update app.py
Browse files
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}/
|
| 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)
|