Update README.md

README.md

@@ -38,11 +38,11 @@ Install package:
 pip install -U textgen
 ```
 
-```shell
->>> from textgen import T5Model
->>> model = T5Model("t5", "shibing624/t5-chinese-couplet")
->>> r = model.predict(["对联：丹枫江冷人初去"])
-['白石矶寒客不归']
+```python
+from textgen import T5Model
+model = T5Model("t5", "shibing624/t5-chinese-couplet")
+r = model.predict(["对联：丹枫江冷人初去"])
+print(r) # ['白石矶寒客不归']
 ```
 
 ## Usage (HuggingFace Transformers)