nagasurendra commited on
Commit
41ba03c
·
verified ·
1 Parent(s): 3863812

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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
+