Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,7 @@ model = transformers.EncoderDecoderModel.from_pretrained("sappho192/ffxiv-ja-ko-
|
|
11 |
def translate(text_src):
|
12 |
embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
|
13 |
embeddings = {k: v for k, v in embeddings.items()}
|
14 |
-
output = model.generate(**embeddings,
|
15 |
-
num_beams=5, num_return_sequences=1, no_repeat_ngram_size=1, remove_invalid_values=True)[0, 1:-1]
|
16 |
text_trg = trg_tokenizer.decode(output.cpu())
|
17 |
return text_trg
|
18 |
|
|
|
11 |
def translate(text_src):
|
12 |
embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
|
13 |
embeddings = {k: v for k, v in embeddings.items()}
|
14 |
+
output = model.generate(**embeddings)[0, 1:-1]
|
|
|
15 |
text_trg = trg_tokenizer.decode(output.cpu())
|
16 |
return text_trg
|
17 |
|