Update models/solar_model.py

models/solar_model.py

@@ -1,8 +1,9 @@
-from ultralytics import YOLO
+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")
+        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 Exception(f"Error loading solar model: {str(e)}")  # Raise error with details