Update README.md
Browse files
README.md
CHANGED
@@ -1,13 +1,13 @@
|
|
1 |
-
---
|
2 |
-
license: mit
|
3 |
-
datasets:
|
4 |
-
- flytech/python-codes-25k
|
5 |
-
tags:
|
6 |
-
- code
|
7 |
-
language:
|
8 |
-
- en
|
9 |
-
library_name: transformers
|
10 |
-
---
|
11 |
# GPT2 PyCode
|
12 |
|
13 |
<!-- Provide a quick summary of what the model is/does. -->
|
@@ -52,6 +52,7 @@ Use the code below to get started with the model.
|
|
52 |
|
53 |
```python
|
54 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
|
|
55 |
import re
|
56 |
|
57 |
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
datasets:
|
4 |
+
- flytech/python-codes-25k
|
5 |
+
tags:
|
6 |
+
- code
|
7 |
+
language:
|
8 |
+
- en
|
9 |
+
library_name: transformers
|
10 |
+
---
|
11 |
# GPT2 PyCode
|
12 |
|
13 |
<!-- Provide a quick summary of what the model is/does. -->
|
|
|
52 |
|
53 |
```python
|
54 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
55 |
+
import torch
|
56 |
import re
|
57 |
|
58 |
|