Spaces:

DSatishchandra
/

solar_wind_fault_detection

Sleeping

DSatishchandra commited on Apr 30

Commit

65afddc

verified ·

1 Parent(s): 23171b2

Update models/windmill_model.py

Files changed (1) hide show

models/windmill_model.py CHANGED Viewed

@@ -1,9 +1,8 @@
-import torch
 def load_windmill_model():
     try:
-        model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, force_reload=True)
-        model.eval()
         return model
     except Exception as e:
         raise Exception(f"Error loading windmill model: {str(e)}")

+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)}")