Update README.md
Browse files
README.md
CHANGED
@@ -20,11 +20,11 @@ japanese-sexual-moderationは、[studio-ousia/luke-japanese-large-lite](https://
|
|
20 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
21 |
import numpy as np
|
22 |
|
23 |
-
|
24 |
|
25 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
26 |
model = AutoModelForSequenceClassification.from_pretrained(
|
27 |
-
|
28 |
problem_type="multi_label_classification",
|
29 |
num_labels=1
|
30 |
)
|
|
|
20 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
21 |
import numpy as np
|
22 |
|
23 |
+
model_id = "oshizo/japanese-sexual-moderation"
|
24 |
|
25 |
+
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
26 |
model = AutoModelForSequenceClassification.from_pretrained(
|
27 |
+
model_id,
|
28 |
problem_type="multi_label_classification",
|
29 |
num_labels=1
|
30 |
)
|