Update README.md

README.md

@@ -6,30 +6,38 @@ tags:
 - gpt2
 - gpt
 license: mit
-datasets:
-- natural questions
-
 widget:
-- text: "Do you like my new haircut?\nperson beta:\n\n"  
-  example_title: "haircut"
-- text: "I love to learn new things.. are you willing to teach me something?\nperson beta:\n\n"  
-  example_title: "teaching"
-- text: "What's your favorite animal? Mine is the dog? \nperson beta:\n\n"
-  example_title: "favorite"
-- text: "how much does it cost?\nperson beta:\n\n"
-  example_title: "money"
-
+- text: |+
+    Do you like my new haircut?
+    person beta:
+
+  example_title: haircut
+- text: |+
+    I love to learn new things.. are you willing to teach me something?
+    person beta:
+
+  example_title: teaching
+- text: |+
+    What's your favorite animal? Mine is the dog? 
+    person beta:
+
+  example_title: favorite
+- text: |+
+    how much does it cost?
+    person beta:
+
+  example_title: money
 inference:
   parameters:
     min_length: 2
     max_length: 64
     length_penalty: 0.6
     no_repeat_ngram_size: 3
-    do_sample: True
+    do_sample: true
     top_p: 0.85
     top_k: 10
     repetition_penalty: 2.1
-    
+pipeline_tag: text-generation
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -92,4 +100,4 @@ The following hyperparameters were used during training:
 
 - Transformers 4.16.1
 - Pytorch 1.10.0+cu111
-- Tokenizers 0.11.0
+- Tokenizers 0.11.0