Update README.md

README.md

@@ -8,11 +8,15 @@ tags:
 - databricks-dolly
 - prompt-category
 language: en
+datasets:
+- databricks/databricks-dolly-15k
+base_model:
+- microsoft/deberta-v3-base
 ---
 
 # 🧠 DeBERTa-v3 Base - Prompt Category Classifier (Fine-tuned)
 
-This model is a fine-tuned version of [`microsoft/deberta-v3-base`](https://huggingface.co/microsoft/deberta-v3-base) on a balanced version of the [databricks-dolly-15k](https://huggingface.co/datasets/databricks/databricks-dolly-15k) dataset.  
+This model is a fine-tuned version of [`microsoft/deberta-v3-base`](https://huggingface.co/microsoft/deberta-v3-base) on the [databricks-dolly-15k](https://huggingface.co/datasets/databricks/databricks-dolly-15k) dataset.  
 It has been trained to classify the **prompt category** based solely on the **response** text.
 
 ## 🗂️ Task
@@ -70,4 +74,4 @@ Apache 2.0
 
 ---
 
-📝 Fine-tuned by [mariadg](https://huggingface.co/mariadg) – for research purposes.
+📝 Fine-tuned by [mariadg](https://huggingface.co/mariadg) – for research purposes.