Update README.md

README.md

@@ -91,9 +91,9 @@ The sketch which can be composed of:
 
 ### How to use the model
 #### 1. If you already have a sketch in mind, and want to get a paragraph based on it...
-```python
+```
 from transformers import pipeline
-# 1. load the model with the huggingface `pipeline`
+# 1. load the model with the huggingface pipeline
 genius = pipeline("text2text-generation", model='beyond/genius-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>"
@@ -101,8 +101,9 @@ sketch = "<mask> Conference on Empirical Methods <mask> submission of research p
 generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
 print(generated_text)
 ```
+
 Output:
-```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.'
 ```