Update README.md

README.md

@@ -1,12 +1,20 @@
 ---
 license: mit
 ---
-Mistral fine-tuned to Multi-NLI (MNLI)
+
+datasets:
+- multi_nli
+language:
+- en
+pipeline_tag: text-classification
+---
+
+# Mistral fine-tuned to Multi-NLI (MNLI
 This model is for Textual Entailment (NLI), i.e., predict whether sentence 1 is supported by sentence 2. It's a 3-way classification where the relationship between sentence 1 and sentence 2 can be entailment, neutral, contradiction.
 
+## Model Details
+- Base model: [Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2)
+- Training data: [MNLI](https://huggingface.co/datasets/multi_nli)
+- Training details: num_epochs = 3, batch_size = 8
 
 
-Model Details
-Base model: Mistral-7B-Instruct-v0.2
-Training data: MNLI
-Training details: num_epochs = 3, batch_size = 8