Minor model card edits

README.md

@@ -13,17 +13,18 @@ tags:
 - jailbreak
 - security
 - moderation
+- prompt-injection
 ---
 
 # Jailbreak Classifier
 
-Classifies prompts as jailbreaks or benign. This is a fine-tune checkpoint of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the [jailbreak-classification](https://huggingface.co/datasets/jackhhao/jailbreak-classification) dataaset.
+Classifies prompts as jailbreaks or benign. This is a fine-tune checkpoint of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the [jailbreak-classification](https://huggingface.co/datasets/jackhhao/jailbreak-classification) dataset.
 
 ## Training Details
 
 ### Training Data
 
-Fine-tuned on the [jailbreak-classification](https://huggingface.co/datasets/jackhhao/jailbreak-classification) dataaset.
+Fine-tuned on the [jailbreak-classification](https://huggingface.co/datasets/jackhhao/jailbreak-classification) dataset.
 
 ### Training Procedure 
 
@@ -31,7 +32,7 @@ Fine-tuned on the [jailbreak-classification](https://huggingface.co/datasets/jac
 
 Fine-tuning hyper-parameters:
 - learning_rate = 5e-5
-- per_device_train_batch_size = 8
-- per_device_eval_batch_size = 8
+- train_batch_size = 8
+- eval_batch_size = 8
 - lr_scheduler_type = linear
 - num_train_epochs = 5.0