Update app.py

app.py

@@ -65,4 +65,16 @@ if __name__ == "__main__":
     
 from datasets import load_dataset
 
-ds = load_dataset("AI-MO/NuminaMath-CoT")
+ds = load_dataset("AI-MO/NuminaMath-CoT")
+from datasets import load_dataset
+
+# 加载数据集
+dataset_name = "AI-MO/NuminaMath-CoT"
+dataset = load_dataset(dataset_name)
+
+# 打印数据集的信息
+print("数据集信息：", dataset)
+
+# 打印前五条训练样本
+print("列名称：", dataset["train"].column_names)
+print("训练集前五条数据：", dataset["train"][:5])