Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|