heegyu commited on
Commit
61473f2
ยท
1 Parent(s): 0c56a01

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -9,8 +9,8 @@ Streamlit Demo: https://heegyukim-open-domain-dialog-st-demo-1tzktp.streamlitapp
9
 
10
  ## ์‚ฌ์šฉ์˜ˆ์‹œ
11
  ```
12
- tokenizer = AutoTokenizer.from_pretrained("heegyu/kodialogpt")
13
- model = AutoModelForCausalLM.from_pretrained("heegyu/kodialogpt")
14
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
15
 
16
  generation_args = dict(
 
9
 
10
  ## ์‚ฌ์šฉ์˜ˆ์‹œ
11
  ```
12
+ tokenizer = AutoTokenizer.from_pretrained("heegyu/kodialogpt-v0")
13
+ model = AutoModelForCausalLM.from_pretrained("heegyu/kodialogpt-v0")
14
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
15
 
16
  generation_args = dict(