Load only COT samples

app.py

@@ -468,7 +468,7 @@ REASONING: This text is polite because it expresses gratitude and encourages the
                         "### Sample Output",
                     )
                     dataset = (
-                        load_dataset("Intel/polite-guard", split="validation")
+                        load_dataset("intel/polite-guard", data_files={"validation": "data/validation/val_cot.csv"})
                         .to_pandas()
                         .sample(n=5)
                     )