Update README.md

README.md

@@ -1,3 +1,14 @@
+---
+language: en
+license: apache-2.0
+tags:
+  - chatbot
+  - dialogpt
+  - causal-lm
+  - transformers
+pipeline_tag: text-generation
+---
+
 # DialoGPT Fine-Tuned Chatbot
 
 This is a fine-tuned version of `microsoft/DialoGPT-small` trained on a small chatbot dataset.
@@ -6,5 +17,5 @@ This is a fine-tuned version of `microsoft/DialoGPT-small` trained on a small ch
 ```python
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="your-username/dialogpt-chatbot")
-print(pipe("Human: Tell me a joke\nAssistant:")[0]["generated_text"])
+chatbot = pipeline("text-generation", model="your-username/dialogpt-chatbot")
+print(chatbot("Human: Tell me a joke\nAssistant:")[0]["generated_text"])