solar_wind_fault_detection_Ver2

Sleeping

DSatishchandra commited on Apr 30

Commit

1e1ee1e

verified ·

1 Parent(s): 0d01b1b

Update models/solar_model.py

Files changed (1) hide show

models/solar_model.py CHANGED Viewed

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

 def load_solar_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 solar model: {str(e)}")