Update README.md

README.md

@@ -31,7 +31,13 @@ model = AutoModelForCausalLM.from_pretrained(
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 input_ids = tokenizer(input_text, return_tensors="pt")
 outputs = model.generate(**input_ids, max_length=128)
-print(tokenizer.decode(outputs[0]))
+output = tokenizer.decode(outputs[0])
+#print("llm output:",output)
+
+backstory=(output.split("\n\n"))[1].split("\n\n")[0]
+goal=(output.split(backstory)[1].replace("<eos>","")).replace("\n\n","")
+print("backstory: ",backstory)
+print("goal: ",goal)
 ```
 # Training Data
 We have 103 rows of descriptions of different roles and their respective goals and backstory which is used to train the models, see [this dataset](https://huggingface.co/datasets/DrDrek/crewai_finetuning_dataset) for details.