tags and üarams

README.md

@@ -1,14 +1,19 @@
 ---
 license: apache-2.0
 tags:
-- generated_from_trainer
+- text-generation
+- chatbot
+- dialogue
+- distilgpt2
+- gpt2
+- ai-msgbot
 
 widget:
 - text: "I know you're tired, but can we go for another walk this evening?\nperson beta:\n\n" 
   example_title: "walk"
 - text: "Have you done anything exciting lately?\nperson beta:\n\n"
   example_title: "activities"
-- text: "hey - do you have a favourite grocery store around here?\nperson beta:\n\n"
+- text: "hey - do you have a favorite grocery store around here?\nperson beta:\n\n"
   example_title: "grocery"
 - text: "Can you take me for dinner somewhere nice this time?\nperson beta:\n\n"
   example_title: "dinner"
@@ -35,8 +40,9 @@ inference:
     no_repeat_ngram_size: 2
     do_sample: True
     top_p: 0.95
-    top_k: 30
-    repetition_penalty: 2.1
+    top_k: 20
+    temperature: 0.5
+    repetition_penalty: 3.5
     
 ---