Update README.md
Browse files
README.md
CHANGED
|
@@ -12,7 +12,6 @@ datasets:
|
|
| 12 |
- c4
|
| 13 |
- beyond/chinese_clean_passages_80m
|
| 14 |
|
| 15 |
-
|
| 16 |
widget:
|
| 17 |
- text: "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 18 |
example_title: "Example 1"
|
|
@@ -32,12 +31,12 @@ inference:
|
|
| 32 |
do_sample: True
|
| 33 |
---
|
| 34 |
|
| 35 |
-
#
|
|
|
|
| 36 |
|
| 37 |
-
**How to use:**
|
| 38 |
```
|
| 39 |
from transformers import pipeline
|
| 40 |
-
genius = pipeline("text2text-generation", model=
|
| 41 |
sketch = "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 42 |
genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
|
| 43 |
```
|
|
|
|
| 12 |
- c4
|
| 13 |
- beyond/chinese_clean_passages_80m
|
| 14 |
|
|
|
|
| 15 |
widget:
|
| 16 |
- text: "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 17 |
example_title: "Example 1"
|
|
|
|
| 31 |
do_sample: True
|
| 32 |
---
|
| 33 |
|
| 34 |
+
# GENIUS: generating text using sketches!
|
| 35 |
+
|
| 36 |
|
|
|
|
| 37 |
```
|
| 38 |
from transformers import pipeline
|
| 39 |
+
genius = pipeline("text2text-generation", model="beyond/genius-large", device=0)
|
| 40 |
sketch = "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 41 |
genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
|
| 42 |
```
|