Update README.md

README.md

@@ -52,7 +52,7 @@ def predict(content_prompt):
                     pad_token_id=tokenizer.convert_tokens_to_ids("<|eot_id|>"),
                 )
         s = generation_output.sequences[0]
-        output = tokenizer.decode(s,skip_special_tokens=True)
+        output = tokenizer.decode(s,skip_special_tokens=False)
         output1 = output.split("<|eot_id|>")[-2].lstrip("<|start_header_id|>assistant<|end_header_id|>").strip()
         # print(output1)
     return output1