solar_wind_fault_detection_Ver2

Sleeping

DSatishchandra commited on Apr 30

Commit

a204ca3

verified ·

1 Parent(s): 9e94818

Update models/solar_model.py

Files changed (1) hide show

models/solar_model.py CHANGED Viewed

@@ -1,5 +1,5 @@
-import torch
 def load_solar_model():
-    model = torch.hub.load("ultralytics/yolov5", "yolov5s")  # Replace with your model path
     return model

+from transformers import AutoModelForObjectDetection
 def load_solar_model():
+    model = AutoModelForObjectDetection.from_pretrained("ultralytics/yolov5s")
     return model