Update README.md
Browse files
README.md
CHANGED
|
@@ -55,7 +55,7 @@ Why write the rest of your email when you can generate it?
|
|
| 55 |
```python
|
| 56 |
from transformers import pipeline
|
| 57 |
|
| 58 |
-
model_tag = "postbot/distilgpt2-emailgen"
|
| 59 |
generator = pipeline(
|
| 60 |
'text-generation',
|
| 61 |
model=model_tag,
|
|
@@ -75,7 +75,7 @@ result = generator(
|
|
| 75 |
print(result[0]['generated_text'])
|
| 76 |
```
|
| 77 |
|
| 78 |
-
This model is a fine-tuned version of
|
| 79 |
It achieves the following results on the evaluation set:
|
| 80 |
- Loss: 1.9126
|
| 81 |
|
|
|
|
| 55 |
```python
|
| 56 |
from transformers import pipeline
|
| 57 |
|
| 58 |
+
model_tag = "postbot/distilgpt2-emailgen-V2"
|
| 59 |
generator = pipeline(
|
| 60 |
'text-generation',
|
| 61 |
model=model_tag,
|
|
|
|
| 75 |
print(result[0]['generated_text'])
|
| 76 |
```
|
| 77 |
|
| 78 |
+
This model is a fine-tuned version of `distilgpt2` on the postbot/multi-emails-100k dataset.
|
| 79 |
It achieves the following results on the evaluation set:
|
| 80 |
- Loss: 1.9126
|
| 81 |
|