Update README.md
Browse files
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(
|