Update README.md

README.md

@@ -36,8 +36,22 @@ SEGA is a general text augmentation model that can be used for data augmentation
 - Github: [this repository](to_be_added). 
 
 
-## Model description
 
+### How to use
+```python
+from transformers import pipeline
+# 1. load the model with the huggingface `pipeline`
+sega = pipeline("text2text-generation", model='beyond/sega-large', device=0)
+# 2. provide a sketch (joint by <mask> tokens)
+sketch = "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
+# 3. just do it!
+generated_text = sega(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
+print(generated_text)
+```
+
+```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.'
+```
 
 ## Model variations
 
@@ -56,7 +70,7 @@ SEGA is a general text augmentation model that can be used for data augmentation
 
 
 
-### How to use
+
 
 
 ### Limitations and bias