Update README.md

README.md

@@ -92,13 +92,13 @@ 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)
+# 2.provide a sketch (joint by <mask> tokens)
 sketch = "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
-# 3. here we go!
+# 3.here we go!
 generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
 print(generated_text)
 ```