engmatic-earth commited on
Commit
3c0b1d8
1 Parent(s): 7efffba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", padding=True))
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(" ", "")