snoop2head commited on
Commit
0dc4a45
·
1 Parent(s): db6849b

fix: decoding function

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ if st.button('문장 변환'):
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,7 +32,7 @@ if st.button('문장 변환'):
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,
 
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,