Update README.md

README.md

@@ -23,7 +23,10 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 # Model
 model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 
-enhancer = pipeline('text2text-generation', model=model, tokenizer=tokenizer, device=device)
+enhancer = pipeline('text2text-generation',
+                    model=model,
+                    tokenizer=tokenizer,
+                    device=device)
 max_target_length = 256
 prefix = "enhance prompt: "