Update KerasNLP quickstart example

Forgot to sync a change to keras.io landing page!
keras-team · Aug 16, 2023 · 0f9362c · 0f9362c
1 parent 4089bcb
commit 0f9362c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/templates/keras_nlp/index.md b/templates/keras_nlp/index.md
@@ -114,7 +114,10 @@ imdb_train, imdb_test = tfds.load(
     batch_size=16,
 )
 # Load a BERT model.
-classifier = keras_nlp.models.BertClassifier.from_preset("bert_base_en_uncased")
+classifier = keras_nlp.models.BertClassifier.from_preset(
+    "bert_base_en_uncased", 
+    num_classes=2,
+)
 # Fine-tune on IMDb movie reviews.
 classifier.fit(imdb_train, validation_data=imdb_test)
 # Predict two new examples.