gait-analysis-backend / download_model.py
techSnipe's picture
Upload folder using huggingface_hub
83ee618 verified
raw
history blame contribute delete
435 Bytes
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()