solar_wind_fault_detection_Ver2

Sleeping

DSatishchandra commited on Apr 30

Commit

2c8c1b3

verified ·

1 Parent(s): 1e1ee1e

Update models/windmill_model.py

Files changed (1) hide show

models/windmill_model.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch
 def load_windmill_model():
     try:
-        model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, force_reload=False)
         model.eval()
         return model
     except Exception as e:

 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: