Commit
·
3c0b1d8
1
Parent(s):
7efffba
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained("engmatic-earth/mt5-zh-ja-en-trimm
|
|
6 |
|
7 |
def output(raw_input_sentence):
|
8 |
target_sentence = ["en2ja: " + str(raw_input_sentence)]
|
9 |
-
translated = model.generate(**tokenizer(target_sentence, return_tensors="pt",
|
10 |
tgt_text = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
|
11 |
translated_phrase = tgt_text[0]
|
12 |
translated_phrase = translated_phrase.replace(" ", "")
|
|
|
6 |
|
7 |
def output(raw_input_sentence):
|
8 |
target_sentence = ["en2ja: " + str(raw_input_sentence)]
|
9 |
+
translated = model.generate(**tokenizer(target_sentence, return_tensors="pt"), max_length = 1000)
|
10 |
tgt_text = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
|
11 |
translated_phrase = tgt_text[0]
|
12 |
translated_phrase = translated_phrase.replace(" ", "")
|