Update README.md with new model card content
Browse files
README.md
CHANGED
@@ -100,7 +100,7 @@ labels = [0, 3]
|
|
100 |
|
101 |
# Pretrained classifier.
|
102 |
classifier = keras_nlp.models.RobertaClassifier.from_preset(
|
103 |
-
"roberta_base_en",
|
104 |
num_classes=4,
|
105 |
)
|
106 |
classifier.fit(x=features, y=labels, batch_size=2)
|
@@ -128,7 +128,7 @@ labels = [0, 3]
|
|
128 |
|
129 |
# Pretrained classifier without preprocessing.
|
130 |
classifier = keras_nlp.models.RobertaClassifier.from_preset(
|
131 |
-
"roberta_base_en",
|
132 |
num_classes=4,
|
133 |
preprocessor=None,
|
134 |
)
|
|
|
100 |
|
101 |
# Pretrained classifier.
|
102 |
classifier = keras_nlp.models.RobertaClassifier.from_preset(
|
103 |
+
"hf://keras/roberta_base_en",
|
104 |
num_classes=4,
|
105 |
)
|
106 |
classifier.fit(x=features, y=labels, batch_size=2)
|
|
|
128 |
|
129 |
# Pretrained classifier without preprocessing.
|
130 |
classifier = keras_nlp.models.RobertaClassifier.from_preset(
|
131 |
+
"hf://keras/roberta_base_en",
|
132 |
num_classes=4,
|
133 |
preprocessor=None,
|
134 |
)
|