Update README.md

README.md

@@ -1,6 +1,7 @@
 ---
 language: en
 tags:
+- SEGA
 - data augmentation
 - keywords-to-text generation
 - sketch-to-text generation
@@ -32,7 +33,7 @@ inference:
 
 **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. 
+**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)
 - Github: [this repository](to_be_added). 
@@ -50,7 +51,7 @@ sketch = "<mask> Conference on Empirical Methods <mask> submission of research p
 generated_text = sega(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
 print(generated_text)
 ```
-
+Output:
 ```shell
 'The Conference on Empirical Methods welcomes the submission of research papers. Abstracts should be in the form of a paper or presentation. Please submit abstracts to the following email address: eemml.stanford.edu. The conference will be held at Stanford University on April 1618, 2019. The theme of the conference is Deep Learning.'
 ```