Update README.md

README.md

@@ -28,7 +28,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 model_name = "nayohan/llama3-8b-translation-en-ko-1sent"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name,device_map="auto",torch_dtype=torch.float16)
+model = AutoModelForCausalLM.from_pretrained(
+  model_name,
+  device_map="auto",
+  torch_dtype=torch.float16
+)
 ```
 
 ### **Generating Text**
@@ -41,7 +45,14 @@ SYSTEM_PROMPT=f"Acts as a translator. Translate {source} sentences into {target}
 s = "The aerospace industry is a flower in the field of technology and science."
 conversation = [{'role': 'system', 'content': SYSTEM_PROMPT},
                 {'role': 'user', 'content': s}]
-inputs = tokenizer.apply_chat_template(conversation,tokenize=True,add_generation_prompt=True,return_tensors='pt').to("cuda")
+
+inputs = tokenizer.apply_chat_template(
+  conversation,
+  tokenize=True,
+  add_generation_prompt=True,
+  return_tensors='pt'
+).to("cuda")
+
 outputs = model.generate(inputs, max_new_tokens=256)
 print(tokenizer.decode(outputs[0][len(inputs[0]):]))
 ```