Update README.md

README.md

@@ -13,8 +13,7 @@ language:
 ```py
 
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
-from transformers import TextStreamer
+from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
 
 model_name = "Writer/palmyra-20b-chat"
 
@@ -52,6 +51,9 @@ if "token_type_ids" in model_inputs:
     del model_inputs["token_type_ids"]
 
 all_inputs = {**model_inputs, **gen_conf}
-_ = model.generate(**all_inputs, streamer=streamer)
+output = model.generate(**all_inputs, streamer=streamer)
+
+print("-"*20)
+print(output)
 
 ```