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,
|