Update app.py

app.py

@@ -304,10 +304,24 @@ print(train_dataset[:2])  # 查看训练集的前两个样本
 # 如果你知道确切的子集名称，也可以直接加载它  
 # train_dataset = load_dataset('fka/awesome-chatgpt-prompts', split='train')
 '''
+#############################
+'''
 from datasets import load_dataset
 dataset = load_dataset("aspnet/yoloensembledata")
 #print(dataset)
 print(dataset['train'])
 print(dataset.keys())
 print(dataset['test'])
-print(dataset['validation'])
+print(dataset['validation'])
+'''
+#########################
+
+from ultralytics import YOLO
+
+# Load a model
+model = YOLO("yolov8n.yaml")  # build a new model from YAML
+model = YOLO("yolov8n.pt")  # load a pretrained model (recommended for training)
+model = YOLO("yolov8n.yaml").load("yolov8n.pt")  # build from YAML and transfer weights
+
+# Train the model
+results = model.train(data="coco8.yaml", epochs=100, imgsz=640)