Update models/yolov8_model.pt

models/yolov8_model.pt

@@ -1,9 +1,14 @@
 # scripts/download_yolov8_model.py
 from ultralytics import YOLO
+model = YOLO('yolov8n.pt')
+# After training
+model.save('models/yolov8_model.pt')
 import os
 import torch
 import requests
 import shutil
+print(torch.__version__)
+print(ultralytics.__version__)
 
 # Directory to save the model
 MODEL_DIR = "models"
@@ -12,6 +17,13 @@ os.makedirs(MODEL_DIR, exist_ok=True)
 # Path to save the model
 MODEL_PATH = os.path.join(MODEL_DIR, "yolov8_model.pt")
 
+MODEL_PATH = 'models/yolov8_model.pt'
+try:
+    checkpoint = torch.load(MODEL_PATH, map_location='cpu')
+    print("Model loaded successfully:", checkpoint.keys())
+except Exception as e:
+    print("Error:", e)
+    
 # URL for yolov8n.pt from Ultralytics GitHub releases
 MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt"