Update README.md

README.md

@@ -149,14 +149,14 @@ GENERATION_CONFIG = GenerationConfig(
     max_new_tokens=1024,
 )
 
- outputs = model.generate(
+outputs = model.generate(
      inputs.input_ids.to("cuda:0"),
      GENERATION_CONFIG,
      attention_mask=inputs.attention_mask.to("cuda:0"),
  )
- outputs = outputs[:, inputs.input_ids.shape[-1]:]
- answers = processor.batch_decode(outputs, skip_special_tokens=True)
- for ans in answers:
-     print(ans)
+outputs = outputs[:, inputs.input_ids.shape[-1]:]
+answers = processor.batch_decode(outputs, skip_special_tokens=True)
+for ans in answers:
+    print(ans)
 
 ```