models/solar_model.py

from ultralytics import YOLO

def load_solar_model():
    try:
        model = YOLO("yolov5s.pt")  # Load the YOLOv5s model
        return model
    except Exception as e:
        raise Exception(f"Error loading solar model: {str(e)}")