Update README.md

README.md

@@ -1,12 +1,17 @@
 ---
 language: en
-license: apache-2.0
+license: mit
 library_name: transformers
 tags:
   - llama
+  - llama-3
   - causal-lm
-  - merged
   - vllm
+  - conversational
+  - cognitive-therapy
+  - mental-health
+  - lora
+  - peft
 inference:
   parameters:
     max_new_tokens: 256
@@ -15,21 +20,25 @@ inference:
     repetition_penalty: 1.1
 ---
 
-# CBT-Copilot 🧠
+# 🧠 CBT-Copilot
 
-CBT-Copilot is a fine-tuned version of `meta-llama/Llama-3.2-3B-Instruct`, designed to simulate conversations for cognitive behavioral therapy (CBT) support. It has been trained using LoRA and merged into a standalone model.
+**CBT-Copilot** is a fine-tuned version of [`meta-llama/Llama-3.2-3B-Instruct`](https://huggingface.co/meta-llama/Llama-3.2-3B-Instruct), designed to simulate compassionate and supportive dialogues in the style of **Cognitive Behavioral Therapy (CBT)**.
 
-The model is now compatible with `transformers`, `vLLM`, and other inference frameworks.
+Fine-tuned using LoRA on the [`Lumiiree/therapod-dpo`](https://huggingface.co/datasets/Lumiiree/therapod-dpo) dataset and merged into a standalone model, it supports deployment through `transformers`, `vLLM`, and other inference frameworks.
 
-## 🚀 How to Use (vLLM)
+---
+
+## 🚀 How to Use (with vLLM)
 
-You can serve it with [vLLM](https://github.com/vllm-project/vllm):
+Serve this model using [vLLM](https://github.com/vllm-project/vllm):
 
 ```bash
-python3 -m vllm.entrypoints.openai.api_server --model your-username/CBT-Copilot
+pip install vllm[serve]
+
+python3 -m vllm.entrypoints.openai.api_server --model thillaic/CBT-Copilot
 ```
 
-Then query it like this:
+Then query it via the OpenAI-compatible API:
 
 ```python
 import openai
@@ -40,6 +49,7 @@ openai.api_base = "http://localhost:8000/v1"
 response = openai.ChatCompletion.create(
     model="CBT-Copilot",
     messages=[
+        {"role": "system", "content": "You are a compassionate CBT therapist."},
         {"role": "user", "content": "I've been feeling really anxious lately. What can I do?"}
     ]
 )
@@ -47,19 +57,33 @@ response = openai.ChatCompletion.create(
 print(response["choices"][0]["message"]["content"])
 ```
 
+---
+
 ## 🧠 Intended Use
 
-This model is intended for educational and prototyping purposes in mental health-related chatbot systems. It is **not a substitute for professional therapy**.
+This model is intended for:
 
-## 📜 License
+- Mental health chatbot research
+- Journaling and self-reflection tools
+- Prototyping conversational CBT agents
 
-This model is licensed under the Apache 2.0 license.
+> ⚠️ **Disclaimer**: This model is not a replacement for licensed mental health professionals. It should only be used for **educational, research, or prototyping purposes**.
 
+---
+
+## 📜 License
+
+Licensed under the **MIT License**.
 
 ---
 
-*Model prepared and fine-tuned by **Thillai Chithambaram***
+## 🙏 Acknowledgements
+
+- Based on Meta’s LLaMA 3.2B Instruct model
+- Trained on [Lumiiree/therapod-dpo](https://huggingface.co/datasets/Lumiiree/therapod-dpo)
+- Fine-tuning performed with Hugging Face `transformers`, `PEFT`, and `LoRA`
 
 ---
 
-**Made by ThillaiC**
+**🛠️ Model developed by [Thillai Chithambaram](https://huggingface.co/thillaic)**
+