Update README.md

README.md

@@ -1,18 +1,56 @@
 ---
 license: bigscience-bloom-rail-1.0
 tags:
+- text generation
 - generated_from_trainer
-model-index:
-- name: bloom-1b1-emailgen-v1
-  results: []
+- email generation
+- email
+datasets:
+- aeslc
+- postbot/multi-emails-100k
+
+widget:
+- text: "Good Morning Professor Beans,
+
+Hope you are doing well. I just wanted to reach out and ask if differential calculus will be on the exam"
+  example_title: "email to prof"
+- text: "Hey <NAME>,\n\nThank you for signing up for my weekly newsletter. Before we get started, you'll have to confirm your email address." 
+  example_title: "newsletter"
+- text: "Hi <NAME>,\n\nI hope this email finds you well. I wanted to reach out and ask about office hours" 
+  example_title: "office hours"
+- text: "Greetings <NAME>,\n\nI hope you had a splendid evening at the Company sausage eating festival. I am reaching out because" 
+  example_title: "festival"
+- text: "Good Morning Harold,\n\nI was wondering when the next" 
+  example_title: "event"
+- text: "URGENT - I need the TPS reports"
+  example_title: "URGENT"
+- text: "Hi Archibald,\n\nI hope this email finds you extremely well." 
+  example_title: "emails that find you"
+- text: "Hello there.\n\nI just wanted to reach out and check in to"
+  example_title: "checking in"
+- text: "Hello <NAME>,\n\nI hope this email finds you well. I wanted to reach out and see if you've enjoyed your time with us"
+  example_title: "work well"
+- text: "Hi <NAME>,\n\nI hope this email finds you well. I wanted to reach out and see if we could catch up"
+  example_title: "catch up"
+- text: "I'm <NAME> and I just moved into the area and wanted to reach out and get some details on where I could get groceries and"
+  example_title: "grocery"
+parameters:
+  min_length: 32
+  max_length: 128
+  no_repeat_ngram_size: 2
+  do_sample: True
+  temperature: 0.3
+  top_k: 20
+  top_p: 0.95
+  repetition_penalty: 3.5
+  length_penalty: 0.9
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
 
 # bloom-1b1-emailgen-v1
 
-This model is a fine-tuned version of [bigscience/bloom-1b1](https://huggingface.co/bigscience/bloom-1b1) on an unknown dataset.
+This model is a fine-tuned version of [bigscience/bloom-1b1](https://huggingface.co/bigscience/bloom-1b1) on the ` postbot/multi-emails-100k` dataset.
+
 It achieves the following results on the evaluation set:
 - Loss: 1.7397