metadata
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
from transformers import pipeline
chatbot = pipeline("text-generation", model="your-username/dialogpt-chatbot")
print(chatbot("Human: Tell me a joke\nAssistant:")[0]["generated_text"])