fix: decoding function

app.py

@@ -23,16 +23,16 @@ if st.button('문장 변환'):
 
     else:
         translated = ko2en_model.generate(
-        **tokenizer(src_text, return_tensors="pt", padding=True, max_length=64,),
+        **tokenizer([src_text], return_tensors="pt", padding=True, max_length=64,),
         max_length=64,
         num_beams=5,
         repetition_penalty=1.3,
         no_repeat_ngram_size=3,
         num_return_sequences=1,
         ) 
-        
+        list_translated = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
         backtranslated = en2ko_model.generate(
-        **tokenizer(translated, return_tensors="pt", padding=True, max_length=64,),
+        **tokenizer(list_translated, return_tensors="pt", padding=True, max_length=64,),
         max_length=64,
         num_beams=5,
         repetition_penalty=1.3,
@@ -43,6 +43,6 @@ else:
     pass
 
 
-print(backtranslated)
+print([tokenizer.decode(t, skip_special_tokens=True) for t in backtranslated])
 
-st.write(backtranslated)
+st.write([tokenizer.decode(t, skip_special_tokens=True) for t in backtranslated][0])