Update README.md

README.md

@@ -23,13 +23,15 @@ which is based on [GPT-2 base Thai](https://huggingface.co/flax-community/gpt2-b
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("Kongfha/PhraAphaiManee-LM")
+model_name = "Kongfha/PhraAphaiManee-LM"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
-model = AutoModelForCausalLM.from_pretrained("Kongfha/PhraAphaiManee-LM")
+generate = pipeline("text-generation",
+                    model=model,
+                    tokenizer=tokenizer)
 
-generate = pipeline("text-generation",model=model,tokenizer=tokenizer)    
 text = "๏ สัมผัสเส้นขอบฟ้าชลาลัย"
-
 generated_text = generate(text,max_length=140,top_k=25,temperature=1) # parameters can be varied 
 
 print(f"Input: {text}")