Update README.md

README.md

@@ -14,7 +14,8 @@ pipeline_tag: text-generation
 
 # Model Card for EventModel-1.2B
 
-This model is a fine-tuned version of [liquidai/LFM2-1.2B](https://huggingface.co/liquidai/LFM2-1.2B).
+EventModel is a 1.2B parameter model finetune of LFM2-1.2B using data extracted from r/parents. The idea is to come up with problems that a kid of certain age would face. This is done by using data from r/Parenting, analyzing the problem, analyzing the kid group using iterative few-shot prompting, then finetunning a generative model with the results.
+
 It has been trained using [TRL](https://github.com/huggingface/trl).
 
 ## Quick start
@@ -22,8 +23,8 @@ It has been trained using [TRL](https://github.com/huggingface/trl).
 ```python
 from transformers import pipeline
 
-question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
-generator = pipeline("text-generation", model="None", device="cuda")
+question = "13 year old, boy"
+generator = pipeline("text-generation", model="mzen/EventModel-1.2B", device="cuda")
 output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
 print(output["generated_text"])
 ```