MegaTronX commited on
Commit
3dabd15
·
verified ·
1 Parent(s): c14f038

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ my_tkn = AutoTokenizer.from_pretrained(mdl_name)
13
  #opus_translator = pipeline("translation", model=mdl_name)
14
 
15
  def translate(text):
16
- inputs = my_tkn(text, return_tensors="pt")
17
  trans_output = mdl.generate(**inputs)
18
  response = my_tkn.decode(trans_output[0], skip_special_tokens=True)
19
 
 
13
  #opus_translator = pipeline("translation", model=mdl_name)
14
 
15
  def translate(text):
16
+ inputs = my_tkn(text, return_tensors="pt", truncation=True, max_length=512)
17
  trans_output = mdl.generate(**inputs)
18
  response = my_tkn.decode(trans_output[0], skip_special_tokens=True)
19