Update README.md

README.md

@@ -14,7 +14,7 @@ HF_TOKEN = "<TOKEN HERE>"
 tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-it", token=HF_TOKEN)
 base_model = AutoModelForSequenceClassification.from_pretrained(
     "google/gemma-2-2b-it",
-    num_labels=5,  # Set the number of labels to 5
+    num_labels=5,
     token=HF_TOKEN,
     id2label={
         0: "prompt_injection",