Update app.py

app.py

@@ -68,13 +68,15 @@ from datasets import load_dataset
 ds = load_dataset("AI-MO/NuminaMath-CoT")
 from datasets import load_dataset
 
-# 加载数据集
-dataset_name = "AI-MO/NuminaMath-CoT"
-dataset = load_dataset(dataset_name)
+from datasets import load_dataset
+import gradio as gr
 
-# 打印数据集的信息
-print("数据集信息：", dataset)
+def show_data():
+    # 加载数据集
+    dataset = load_dataset("AI-MO/NuminaMath-CoT")
+    train_data = dataset["train"][:5]
+    return train_data
 
-# 打印前五条训练样本
-print("列名称：", dataset["train"].column_names)
-print("训练集前五条数据：", dataset["train"][:5])
+# 使用 Gradio 界面显示测试数据
+demo = gr.Interface(fn=show_data, inputs=None, outputs="text", title="数据集测试")
+demo.launch()