pszemraj commited on
Commit
8ff0b1c
1 Parent(s): f7449c7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -6
README.md CHANGED
@@ -41,11 +41,10 @@ Why write the rest of your email when you can generate it?
41
  ```python
42
  from transformers import pipeline
43
 
44
- model_tag = "pszemraj/gpt2-medium-email-generation"
45
  generator = pipeline(
46
  'text-generation',
47
  model=model_tag,
48
- use_fast=False,
49
  do_sample=False,
50
  early_stopping=True,
51
  )
@@ -77,10 +76,6 @@ It achieves the following results on the evaluation set:
77
 
78
  More information needed
79
 
80
- ## Training and evaluation data
81
-
82
- More information needed
83
-
84
  ## Training procedure
85
 
86
  ### Training hyperparameters
 
41
  ```python
42
  from transformers import pipeline
43
 
44
+ model_tag = "postbot/distilgpt2-emailgen"
45
  generator = pipeline(
46
  'text-generation',
47
  model=model_tag,
 
48
  do_sample=False,
49
  early_stopping=True,
50
  )
 
76
 
77
  More information needed
78
 
 
 
 
 
79
  ## Training procedure
80
 
81
  ### Training hyperparameters