Update models/yolov8_model.pt

models/yolov8_model.pt

@@ -10,8 +10,9 @@ os.makedirs(MODEL_DIR, exist_ok=True)
 MODEL_PATH = os.path.join(MODEL_DIR, "yolov8_model.pt")
 
 # Download pre-trained YOLOv8 nano model
-model = YOLO("yolov8n.pt")  # Automatically downloads from Ultralytics
-
-# Save the model to models/yolov8_model.pt
-model.save(MODEL_PATH)
-print(f"Model saved to {MODEL_PATH}")
+try:
+    model = YOLO("yolov8n.pt")  # Automatically downloads from Ultralytics
+    model.save(MODEL_PATH)
+    print(f"Model saved to {MODEL_PATH}")
+except Exception as e:
+    print(f"Failed to download or save model: {str(e)}")