Update README.md

README.md

@@ -1,3 +1,13 @@
+---
+license: cc-by-nc-4.0
+datasets:
+- mediabiasgroup/BABE
+language:
+- en
+base_model:
+- FacebookAI/roberta-base
+pipeline_tag: text-classification
+---
 Here’s a template for a `README.md` file that you can reuse for each of your models on Hugging Face. It is designed to provide a comprehensive overview of the model, its usage, links to relevant papers, datasets, and results:
 
 ---
@@ -13,21 +23,6 @@ Here’s a template for a `README.md` file that you can reuse for each of your m
 
 ---
 
-## Table of Contents
-
-1. [Model Overview](#model-overview)  
-2. [Model Architecture](#model-architecture)  
-3. [Training Data](#training-data)  
-4. [Evaluation Results](#evaluation-results)  
-5. [Usage](#usage)  
-6. [Example Code](#example-code)  
-7. [Related Papers](#related-papers)  
-8. [Datasets](#datasets)  
-9. [Limitations](#limitations)  
-10. [Citation](#citation)
-
----
-
 ## Model Overview
 
 This model is a [token-level/sentence-level/paragraph-level] classifier that was trained for [specific task, e.g., sentiment analysis, named entity recognition, etc.]. The model is based on [model architecture, e.g., BERT, RoBERTa, etc.] and has been fine-tuned on [mention the dataset] for [number of epochs or other training details].
@@ -36,17 +31,13 @@ It achieves state-of-the-art performance on [mention dataset or task] and is spe
 
 ---
 
-## Model Architecture
+## Training details
 
 - **Base Model:** [mention architecture, e.g., BERT-base, RoBERTa-large, etc.]  
 - **Number of Parameters:** [number of parameters]  
-- **Layers:** [number of layers, if applicable]  
-- **Attention Heads:** [number of attention heads, if applicable]  
-- **Max Sequence Length:** [max input length, if relevant]  
+- **Max Sequence Length:** [max input length, if relevant]
 
----
-
-## Training Data
+### Training Data
 
 The model was fine-tuned on the [name of dataset] dataset. This dataset consists of [short description of dataset, e.g., number of instances, labels, any important data characteristics].
 
@@ -132,16 +123,6 @@ Please cite this paper if you use the model.
 
 ---
 
-## Datasets
-
-The model was trained on the following dataset(s):
-
-- **Dataset Name:** [Dataset Name](dataset_url)  
-  **Size:** [Dataset size]  
-  **Number of Labels:** [Number of labels or classes]  
-  **Availability:** [Open-source or proprietary]
-
----
 
 ## Limitations