Update README.md
Browse files
README.md
CHANGED
|
@@ -46,7 +46,7 @@ You can use this model directly with a pipeline for masked language modeling:
|
|
| 46 |
from transformers import pipeline
|
| 47 |
# 1. load the model with the huggingface `pipeline`
|
| 48 |
genius = pipeline("text2text-generation", model='beyond/genius-large', device=0)
|
| 49 |
-
# 2. provide a sketch (joint by mask tokens)
|
| 50 |
sketch = "your_sketch"
|
| 51 |
# 3. here we go!
|
| 52 |
generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
|
|
|
|
| 46 |
from transformers import pipeline
|
| 47 |
# 1. load the model with the huggingface `pipeline`
|
| 48 |
genius = pipeline("text2text-generation", model='beyond/genius-large', device=0)
|
| 49 |
+
# 2. provide a sketch (joint by <mask> tokens)
|
| 50 |
sketch = "your_sketch"
|
| 51 |
# 3. here we go!
|
| 52 |
generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
|