beyond commited on
Commit
2693249
·
1 Parent(s): 0843263

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -44,12 +44,10 @@ inference:
44
  You can use this model directly with a pipeline for masked language modeling:
45
 
46
  ```python
47
- from transformers import pipeline
48
- # 1. load the model with the huggingface `pipeline`
49
- genius = pipeline("text2text-generation", model='beyond/genius-large', device=0)
50
- # 2. provide a sketch (joint by <mask> tokens)
51
- sketch = "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
52
- # 3. here we go!
53
- generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
54
- print(generated_text)
55
  ```
 
44
  You can use this model directly with a pipeline for masked language modeling:
45
 
46
  ```python
47
+ from transformers import BertTokenizer, TFBertModel
48
+ tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
49
+ model = TFBertModel.from_pretrained("bert-base-uncased")
50
+ text = "Replace me by any text you'd like."
51
+ encoded_input = tokenizer(text, return_tensors='tf')
52
+ output = model(encoded_input)
 
 
53
  ```