Update models/windmill_model.py

models/windmill_model.py

@@ -1,5 +1,5 @@
-import torch
+from transformers import AutoModelForObjectDetection
 
 def load_windmill_model():
-    model = torch.hub.load("ultralytics/yolov5", "yolov5s")  # Replace with your windmill model path
+    model = AutoModelForObjectDetection.from_pretrained("ultralytics/yolov5s")  # You may want to customize this model
     return model