Update app.py

app.py

@@ -120,7 +120,7 @@ def n_line_poem(input_letter):
     poem_dict = {}
 
     for letter, res in zip(input_letter, res_l):
-        decode_res = tokenizer.decode(res, clean_up_tokenization_spaces=True)
+        decode_res = tokenizer.decode(res, clean_up_tokenization_spaces=True, skip_special_tokens=True)
         poem_dict[poem(letter)] = decode_res
 
     return poem_dict
@@ -186,5 +186,4 @@ with row2_1:
             result = n_line_poem(word_input)
         st.success('완료됐습니다!')
         for r in result:
-            st.write(f'{r} : {result[r]}')
-
+            st.write(f'{r} : {result[r]}')