Update README.md

README.md

@@ -1,21 +1,48 @@
 ---
-license: apache-2.0
+license: cc-by-nc-4.0
 base_model: google/flan-t5-base
+datasets:
+  - grammarly/coedit
 tags:
 - generated_from_trainer
+- text-generation-inference
 metrics:
 - rouge
 model-index:
-- name: coedit-base
+- name: coedit-small
   results: []
+language:
+- en  
+widget:
+  - text: >-
+      Fix the grammar: When I grow up, I start to understand what he said is
+      quite right.
+    example_title: Fluency
+  - text: >-
+      Make this text coherent: Their flight is weak. They run quickly through
+      the tree canopy.
+    example_title: Coherence
+  - text: >-
+      Rewrite to make this easier to understand: A storm surge is what
+      forecasters consider a hurricane's most treacherous aspect.
+    example_title: Simplification
+  - text: 'Paraphrase this: Do you know where I was born?'
+    example_title: Paraphrase
+  - text: >-
+      Write this more formally: omg i love that song im listening to it right
+      now
+    example_title: Formalize
+  - text: 'Write in a more neutral way: The authors'' exposé on nutrition studies.'
+    example_title: Neutralize  
 ---
 
 <!-- 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. -->
 
-# coedit-base
+# coedit-small
+
+This model is a fine-tuned version of [google/flan-t5-base](https://huggingface.co/google/flan-t5-base) on the [CoEdIT dataset](https://huggingface.co/datasets/grammarly/coedit).
 
-This model is a fine-tuned version of [google/flan-t5-base](https://huggingface.co/google/flan-t5-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.5978
 - Rouge1: 60.5931