solar_wind_fault_detection / models /windmill_model.py
DSatishchandra's picture
Update models/windmill_model.py
65afddc verified
raw
history blame contribute delete
241 Bytes
from ultralytics import YOLO
def load_windmill_model():
try:
model = YOLO("yolov5s.pt") # Load the YOLOv5s model
return model
except Exception as e:
raise Exception(f"Error loading windmill model: {str(e)}")