Update README.md
Browse files
README.md
CHANGED
@@ -13,13 +13,13 @@ pipeline_tag: text-generation
|
|
13 |
japanese-gpt-1b-PII-masking は、 [日本語事前学習済み1B GPTモデル](https://huggingface.co/rinna/japanese-gpt-1b)をベースとして、日本語の文章から個人情報をマスキングするように学習したモデルです。
|
14 |
|
15 |
# Usage
|
16 |
-
```
|
17 |
import torch
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
19 |
|
20 |
input_text = ""
|
21 |
|
22 |
-
model_name = ""
|
23 |
model = AutoModelForCausalLM.from_pretrained(best_model_path)
|
24 |
tokenizer = AutoTokenizer.from_pretrained(best_model_path)
|
25 |
|
|
|
13 |
japanese-gpt-1b-PII-masking は、 [日本語事前学習済み1B GPTモデル](https://huggingface.co/rinna/japanese-gpt-1b)をベースとして、日本語の文章から個人情報をマスキングするように学習したモデルです。
|
14 |
|
15 |
# Usage
|
16 |
+
```python
|
17 |
import torch
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
19 |
|
20 |
input_text = ""
|
21 |
|
22 |
+
model_name = "cameltech/japanese-gpt-1b-PII-masking"
|
23 |
model = AutoModelForCausalLM.from_pretrained(best_model_path)
|
24 |
tokenizer = AutoTokenizer.from_pretrained(best_model_path)
|
25 |
|