Update app.py
Browse files
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=
|
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 |
|