Update README.md

README.md

@@ -92,9 +92,12 @@ checkpoint = "lambdasec/santafixer"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-model = AutoModelForCausalLM.from_pretrained(checkpoint, trust_remote_code=True).to(device)
+model = AutoModelForCausalLM.from_pretrained(checkpoint,
+              trust_remote_code=True).to(device)
 
-input_text = "<fim-prefix>def print_hello_world():\n    <fim-suffix>\n    print('Hello world!')<fim-middle>"
+input_text = "<fim-prefix>def print_hello_world():\n
+              <fim-suffix>\n print('Hello world!')
+              <fim-middle>"
 inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
 outputs = model.generate(inputs)
 print(tokenizer.decode(outputs[0]))