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 |
+
|