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