Update README.md

README.md

@@ -35,6 +35,9 @@ A: 쇼핑하러 갈까? B: 응 좋아. A: 언제 갈까? B:
 
 inputs = [text]
 
+tokenizer = AutoTokenizer.from_pretrained(model_dir)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_dir)
+
 inputs = tokenizer(inputs, max_length=max_input_length, truncation=True, return_tensors="pt")
 output = model.generate(**inputs, num_beams=3, do_sample=True, min_length=20, max_length=500, num_return_sequences=3)
 for i in range(3):