Update load_data.py

load_data.py

@@ -23,8 +23,8 @@ RG_DATASET_NAME = "translated-german-alpaca"
 # The name of the Hub dataset to push the validations every 20 min and keep the dataset synced
 HUB_DATASET_NAME = os.environ.get('HUB_DATASET_NAME', f"{SOURCE_DATASET}_validation")
 
-# The labels for the tasks (they can be extended if needed)
-LABELS = ["BAD INSTRUCTION", "BAD INPUT", "BAD OUTPUT", "INAPPROPRIATE", "BIASED", "ALL GOOD"]
+# The labels for the task (they can be extended if needed)
+LABELS = ["BAD INSTRUCTION", "INAPPROPRIATE", "ALL GOOD", "NOT SURE", "WRONG LANGUAGE"]
 
 @listener(
     dataset=RG_DATASET_NAME,