xujinheng666 commited on
Commit
c1fa9f7
·
verified ·
1 Parent(s): 18d5ab3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ def translate(text):
74
  translations = []
75
  for sentence in sentences:
76
  inputs = tokenizer(sentence, return_tensors="pt").to(device)
77
- outputs = model.generate(inputs["input_ids"], max_length=1000, num_beams=5)
78
  translations.append(tokenizer.decode(outputs[0], skip_special_tokens=True))
79
  return " ".join(translations)
80
 
 
74
  translations = []
75
  for sentence in sentences:
76
  inputs = tokenizer(sentence, return_tensors="pt").to(device)
77
+ outputs = model.generate(inputs["input_ids"], max_length=2000, num_beams=5) # Increased max_length to 2000
78
  translations.append(tokenizer.decode(outputs[0], skip_special_tokens=True))
79
  return " ".join(translations)
80