Update README.md

README.md

@@ -1,13 +1,13 @@
----
-license: mit
-datasets:
-- flytech/python-codes-25k
-tags:
-- code
-language:
-- en
-library_name: transformers
----
+---
+license: mit
+datasets:
+- flytech/python-codes-25k
+tags:
+- code
+language:
+- en
+library_name: transformers
+---
 # GPT2 PyCode
 
 <!-- Provide a quick summary of what the model is/does. -->
@@ -52,6 +52,7 @@ Use the code below to get started with the model.
 
 ```python
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import torch
 import re