ultralytics How to use vivekvar/helmet-v4 with ultralytics:
# Couldn't find a valid YOLO version tag.
# Replace XX with the correct version.
from ultralytics import YOLOvXX
model = YOLOvXX.from_pretrained("vivekvar/helmet-v4")
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True) 
