pszemraj commited on
Commit
80a2aa6
1 Parent(s): 13f0784

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 [postbot/distilgpt2-emailgen-V2](https://huggingface.co/postbot/distilgpt2-emailgen-V2) on the postbot/multi-emails-100k dataset.
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