Update README.md

README.md

@@ -20,11 +20,13 @@ model.to('cuda')
 tokenizer = AutoTokenizer.from_pretrained('MrBananaHuman/kogpt_6b_fp16')
 
 input_text = '이순신은'
-input_ids = tokenizer(input_text, return_tensors='pt').to('cuda')
+input_ids = tokenizer(input_text, return_tensors='pt').input_ids.to('cuda')
 
-output = model.generate(**input_ids)
+output = model.generate(input_ids)
 print(tokenizer.decode(output[0]))
 
+>>> 이순신은 우리에게 무엇인가? 1. 머리말 이글은 임진왜란 당시 이순인이 보여준
+
 ```
 
 ### 참고 링크