Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,3 +10,6 @@ model = YOLO("yolo11n.yaml") # You can choose a different model type like yolo5
|
|
10 |
|
11 |
# Train the model
|
12 |
results = model.train(data=config_path, epochs=1)
|
|
|
|
|
|
|
|
10 |
|
11 |
# Train the model
|
12 |
results = model.train(data=config_path, epochs=1)
|
13 |
+
print("Model saved to:", 'runs/detect/train/weights/best.pt')
|
14 |
+
model.save('path_to_save/best.pt')
|
15 |
+
|