Update README.md

README.md

@@ -19,4 +19,23 @@ tags:
 
 This llama model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+The model is trained on reddit_depression_dataset, The epoch = 1. 
+
+The training is in dialog format, but the user's input is ignored.
+
+For example
+
+```python
+def formatting_prompts_func(examples):
+    texts_dataset = examples['text']
+    formatted_prompts = []
+    for text in texts_dataset:
+        dialog = [
+            {"role": "system", "content": "You are a patient undergoing depression."},
+            # {"role": "user", "content": ""},
+            {"role": "assistant", "content": text}
+        ]
+        formatted_prompt = tokenizer.apply_chat_template(dialog, tokenize=False, add_generation_prompt=False)
+        formatted_prompts.append(formatted_prompt)
+    return {"text": formatted_prompts}
+```