dialogpt-chatbot / README.md
MahiH's picture
Update README.md
0cc404e verified
---
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.
## Example usage
```python
from transformers import pipeline
chatbot = pipeline("text-generation", model="your-username/dialogpt-chatbot")
print(chatbot("Human: Tell me a joke\nAssistant:")[0]["generated_text"])