from ultralytics import YOLO
import os

def download_yolo_model():
    print("Checking and downloading YOLO11 model...")
    try:
        # This will automatically download the model if not present
        model = YOLO("yolo11n-pose.pt")
        print("Model yolo11n-pose.pt is ready!")
    except Exception as e:
        print(f"Error initializing YOLO model: {e}")

if __name__ == "__main__":
    download_yolo_model()