add usage section
Browse files
README.md
CHANGED
@@ -31,7 +31,16 @@ any harmful content generated by GPT2 should not be used in anywhere.
|
|
31 |
|
32 |
## Training and evaluation data
|
33 |
training and validation done on goodreads dataset LABR 80% for trainng and 20% for testing
|
34 |
-
##
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
### Training hyperparameters
|
37 |
|
|
|
31 |
|
32 |
## Training and evaluation data
|
33 |
training and validation done on goodreads dataset LABR 80% for trainng and 20% for testing
|
34 |
+
## Usage
|
35 |
+
|
36 |
+
```
|
37 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
38 |
+
|
39 |
+
tokenizer = AutoTokenizer.from_pretrained("mofawzy/argpt2-goodreads")
|
40 |
+
|
41 |
+
model = AutoModelForCausalLM.from_pretrained("mofawzy/argpt2-goodreads")
|
42 |
+
|
43 |
+
```
|
44 |
|
45 |
### Training hyperparameters
|
46 |
|