jinymusim commited on
Commit
25828f3
1 Parent(s): e3600c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -1
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("gpt")
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