DSatishchandra commited on
Commit
6408e06
·
verified ·
1 Parent(s): 40cf333

Create models/solar_model.py

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