DSatishchandra commited on
Commit
93dbe84
·
verified ·
1 Parent(s): 6408e06

Create windmill_model.py

Browse files
Files changed (1) hide show
  1. models/windmill_model.py +8 -0
models/windmill_model.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from ultralytics import YOLO
2
+
3
+ def load_windmill_model():
4
+ try:
5
+ model = YOLO("yolov5s.pt")
6
+ return model
7
+ except Exception as e:
8
+ raise Exception(f"Error loading windmill model: {str(e)}")