models/solar_model.py

from transformers import AutoModelForObjectDetection

def load_solar_model():
    model = AutoModelForObjectDetection.from_pretrained("ultralytics/yolov5s")
    return model