MahiH commited on
Commit
79d1012
·
verified ·
1 Parent(s): 950b9ab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -8
README.md CHANGED
@@ -1,8 +1,10 @@
1
- ---
2
- license: apache-2.0
3
- language:
4
- - en
5
- base_model:
6
- - microsoft/DialoGPT-small
7
- pipeline_tag: text-generation
8
- ---
 
 
 
1
+ # DialoGPT Fine-Tuned Chatbot
2
+
3
+ This is a fine-tuned version of `microsoft/DialoGPT-small` trained on a small chatbot dataset.
4
+
5
+ ## Example usage
6
+ ```python
7
+ from transformers import pipeline
8
+
9
+ pipe = pipeline("text-generation", model="your-username/dialogpt-chatbot")
10
+ print(pipe("Human: Tell me a joke\nAssistant:")[0]["generated_text"])