Update README.md

README.md

@@ -1,6 +1,16 @@
+---
+language: en
+tags:
+- augmentation
+license: apache-2.0
+datasets:
+- C4
+---
+
 # SEGA-large model
 
 SEGA: SkEtch-based Generative Augmentation
+
 SEGA is a general text augmentation model that can be used for data augmentation for various NLP tasks (including sentiment analysis, topic classification, NER, and QA). SEGA uses an encoder-decoder structure (based on the BART architecture) and is pre-trained on the C4-realnewslike corpus. 
 
 - Paper: [this paper](to_be_added)
@@ -15,12 +25,12 @@ SEGA is a general text augmentation model that can be used for data augmentation
 
 | Model | #params | Language |
 |------------------------|--------------------------------|-------|
-| [`sega-large`](https://huggingface.co/bert-base-uncased) | xM   | English |
-| [`sega-base`](https://huggingface.co/bert-large-uncased)  | xM    | English |
-| [`sega-small`](https://huggingface.co/bert-base-cased)        | xM    | English |
-| [`sega-large-chinese`](https://huggingface.co/bert-large-cased) | xM    |  Chinese |
-| [`sega-base-chinese`](https://huggingface.co/bert-base-chinese) | xM    | Chinese |
-| [`sega-small-chinese`](https://huggingface.co/bert-base-multilingual-cased) | xM | Chinese |
+| [`sega-large`]() | xM   | English |
+| [`sega-base`]()  | xM    | English |
+| [`sega-small`]()        | xM    | English |
+| [`sega-large-chinese`]() | xM    |  Chinese |
+| [`sega-base-chinese`]() | xM    | Chinese |
+| [`sega-small-chinese`]() | xM | Chinese |
 
 
 ## Intended uses & limitations