Update app.py

app.py

@@ -281,7 +281,8 @@ for epoch in range(num_epochs):
 # 保存模型（可选）  
 torch.save(model.state_dict(), 'model.pth')
 '''
-
+####################################
+'''
 from datasets import load_dataset  
   
 # 加载数据集  
@@ -289,3 +290,16 @@ dataset = load_dataset('glue', 'sst2')  # 这里的'sst2'是GLUE数据集下的
   
 # 查看数据集内容  
 print(dataset['train'][:2])  # 查看训练集的前两个样本
+'''
+
+from datasets import load_dataset  	  
+# 加载数据集  
+dataset = load_dataset('fka/awesome-chatgpt-prompts')  
+ # 查看数据集的子集  
+print(dataset.keys())  # 这将输出数据集中所有可用的子集名称，例如：dict_keys(['train', 'validation', 'test']) 
+ # 访问特定子集的数据 
+train_dataset = dataset['train']  
+print(train_dataset[:2])  # 查看训练集的前两个样本
+# 如果你知道确切的子集名称，也可以直接加载它  
+# train_dataset = load_dataset('fka/awesome-chatgpt-prompts', split='train')
+