Update README.md

README.md

@@ -23,10 +23,25 @@ It has been trained using [TRL](https://github.com/huggingface/trl).
 ```python
 from transformers import pipeline
 
-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"])
+pipe = pipeline(
+    "text-generation", 
+    model="mzen/EventModel-1.2B", 
+    trust_remote_code=True,
+    device_map="auto"
+)
+
+prompt = "### Character: 13 year old, boy\n\n### Problem:"
+
+output = pipe(
+    prompt,
+    max_new_tokens=512,
+    do_sample=True,
+    temperature=0.7,
+    top_p=0.9,
+    return_full_text=False 
+)
+
+print(output[0]['generated_text'])
 ```
 
 ## Training procedure