Update README.md

README.md

@@ -80,7 +80,9 @@ model.to(device)
 model.eval()
 
 # Generate
-prompt = "It was a bright cold day in April, and the clocks were striking thirteen. Winston Smith,"
+prompt = (
+"It was a bright cold day in April, and the clocks were striking thirteen. Winston Smith,"
+)
 input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
 output = model.generate(
     input_ids=input_ids,
@@ -93,7 +95,7 @@ output = model.generate(
     repetition_penalty=1.025
 )
 
-# Output the text.
+# Output the text
 print("Prompt:", prompt)
 print("*" * 3)
 for i, sentence in enumerate(output):