Update README.md
Browse files
README.md
CHANGED
|
@@ -1,8 +1,10 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 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"])
|