Spaces:
Runtime error
Runtime error
Commit
·
0dc4a45
1
Parent(s):
db6849b
fix: decoding function
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ if st.button('문장 변환'):
|
|
| 23 |
|
| 24 |
else:
|
| 25 |
translated = ko2en_model.generate(
|
| 26 |
-
**tokenizer(
|
| 27 |
max_length=64,
|
| 28 |
num_beams=5,
|
| 29 |
repetition_penalty=1.3,
|
|
@@ -32,7 +32,7 @@ if st.button('문장 변환'):
|
|
| 32 |
)
|
| 33 |
|
| 34 |
backtranslated = en2ko_model.generate(
|
| 35 |
-
**tokenizer(
|
| 36 |
max_length=64,
|
| 37 |
num_beams=5,
|
| 38 |
repetition_penalty=1.3,
|
|
|
|
| 23 |
|
| 24 |
else:
|
| 25 |
translated = ko2en_model.generate(
|
| 26 |
+
**tokenizer(src_text, return_tensors="pt", padding=True, max_length=64,),
|
| 27 |
max_length=64,
|
| 28 |
num_beams=5,
|
| 29 |
repetition_penalty=1.3,
|
|
|
|
| 32 |
)
|
| 33 |
|
| 34 |
backtranslated = en2ko_model.generate(
|
| 35 |
+
**tokenizer(translated, return_tensors="pt", padding=True, max_length=64,),
|
| 36 |
max_length=64,
|
| 37 |
num_beams=5,
|
| 38 |
repetition_penalty=1.3,
|