Update README.md

README.md

@@ -64,6 +64,7 @@ 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")
+model.to("cuda")
 
 template = """<|im_start|>system
 <|im_end|>
@@ -72,7 +73,7 @@ template = """<|im_start|>system
 <|im_start|>assistant
 """
 
-input = template.format(prompt="Kados pundi kulo saged nyinaoni Basa Jawa kanthi sae?"
+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, streamer= TextStreamer(tokenizer), temperature=.5, use_cache=False, do_sample=True)
 ```