Update README.md
Browse files
README.md
CHANGED
@@ -12,15 +12,21 @@ pipeline_tag: conversational
|
|
12 |
Basic Dialog Model from DialoGPT-small.
|
13 |
Finetuned on Dialog dataset. (Daily Dialog, MultiWoz)
|
14 |
|
|
|
|
|
15 |
### How to use
|
16 |
|
|
|
|
|
|
|
|
|
17 |
Use it as any torch python Language Model
|
18 |
|
19 |
```python
|
20 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
21 |
import torch
|
22 |
|
23 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
24 |
model = AutoModelForCausalLM.from_pretrained("jinymusim/dialogmodel")
|
25 |
|
26 |
# Take user Input
|
|
|
12 |
Basic Dialog Model from DialoGPT-small.
|
13 |
Finetuned on Dialog dataset. (Daily Dialog, MultiWoz)
|
14 |
|
15 |
+
For better usage. Use repo https://github.com/jinymusim/Daily-Dialog-GPT
|
16 |
+
|
17 |
### How to use
|
18 |
|
19 |
+
If used with repo https://github.com/jinymusim/Daily-Dialog-GPT
|
20 |
+
User only needs to start the `ds.py` script. Otherwise use following
|
21 |
+
|
22 |
+
|
23 |
Use it as any torch python Language Model
|
24 |
|
25 |
```python
|
26 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
27 |
import torch
|
28 |
|
29 |
+
tokenizer = AutoTokenizer.from_pretrained("jinymusim/dialogmodel")
|
30 |
model = AutoModelForCausalLM.from_pretrained("jinymusim/dialogmodel")
|
31 |
|
32 |
# Take user Input
|