Update README.md
Browse files
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 = "
|
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
|