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 |
|