Divyasreepat commited on
Commit
1288f84
·
verified ·
1 Parent(s): 677f63f

Update README.md with new model card content

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
  )