models/solar_model.py

from ultralytics import YOLO  # Import YOLO from Ultralytics

# Function to load the YOLOv5 model for solar panels
def load_solar_model():
    try:
        model = YOLO("yolov5s.pt")  # Load pre-trained YOLOv5s model
        return model
    except Exception as e:
        raise Exception(f"Error loading solar model: {str(e)}")  # Raise error with details