Update README.md

README.md

@@ -60,7 +60,7 @@ This is the first version of Bakpia.
 
 ## Use
 ```
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
 
 tokenizer = AutoTokenizer.from_pretrained("afrizalha/Bakpia-V1-1.5B-Javanese")
 model = AutoModelForCausalLM.from_pretrained("afrizalha/Bakpia-V1-1.5B-Javanese")
@@ -74,7 +74,7 @@ template = """<|im_start|>system
 
 input = template.format(prompt="Kados pundi kulo saged nyinaoni Basa Jawa kanthi sae?"
 input = tokenizer([input], return_tensors = "pt").to("cuda")
-outputs = model.generate(**input, max_new_tokens = 1024, temperature=.5, use_cache=False, do_sample=True)
+outputs = model.generate(**input, max_new_tokens = 1024, streamer= TextStreamer(tokenizer), temperature=.5, use_cache=False, do_sample=True)
 print(tokenizer.batch_decode(outputs)[0])
 ```