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 |
```
|