Update README.md

README.md

@@ -158,8 +158,9 @@ with torch.cuda.amp.autocast():
         eos_token_id=tokenizer.eos_token_id,
     )
 
-# Inspect outputs
-print('\n\n', tokenizer.decode(output_tokens[0], skip_special_tokens=True))
+generated_text = tokenizer.decode(output_tokens[0], skip_special_tokens=True)
+# Inspect message response in the outputs
+print(generated_text.split("<human>: ")[1].split("<bot>: ")[-1])
 ```
 
 ## Reproducibility