Update app.py

app.py

@@ -467,11 +467,9 @@ REASONING: This text is polite because it expresses gratitude and encourages the
                     gr.Markdown(
                         "### Sample Output",
                     )
-                    dataset = (
-                        load_dataset("intel/polite-guard", data_files={"validation": "data/validation/val_cot.csv"})
-                        .to_pandas()
-                        .sample(n=5)
-                    )
+                    dataset = load_dataset("intel/polite-guard", data_files={"validation": "data/validation/val_cot.csv"})
+                    dataset = dataset["validation"].to_pandas().sample(n=5)
+
                     dataframe = gr.Dataframe(value=dataset, show_label=False)
 
                 state = gr.State({"output_path": None, "timestamp": None})