mofawzy commited on
Commit
0d3d7f9
1 Parent(s): fa10c2d

add usage section

Browse files
Files changed (1) hide show
  1. README.md +10 -1
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
- ## Training procedure
 
 
 
 
 
 
 
 
 
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