Update README.md

README.md

@@ -9,10 +9,9 @@ The dataset utilized can be found on the *Files and versions* tab under the name
 import torch
 from transformers import BartForConditionalGeneration, AutoTokenizer
 
-model = BartForConditionalGeneration.from_pretrained('guialfaro/korean-paraphrasing')
-tokenizer = AutoTokenizer.from_pretrained('guialfaro/korean-paraphrasing')
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = model.to(device)
+model = BartForConditionalGeneration.from_pretrained('guialfaro/korean-paraphrasing').to(device)
+tokenizer = AutoTokenizer.from_pretrained('guialfaro/korean-paraphrasing')
 
 sentence = "7층 방문을 위해 방문록 작성이 필요합니다."
 text =  f"paraphrase: {sentence} "
@@ -39,7 +38,7 @@ generated_ids = model.generate(
 
 preds = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True) for g in generated_ids]
 
-print (f"Original Sentence :: {sentence}")
-print (f"Paraphrased Sentences :: {preds[0]}")
+print(f"Original Sentence :: {sentence}")
+print(f"Paraphrased Sentence :: {preds[0]}")
 
 ```