Update models/windmill_model.py

models/windmill_model.py

@@ -1,5 +1,9 @@
-from transformers import AutoModelForObjectDetection
+import torch
 
 def load_windmill_model():
-    model = AutoModelForObjectDetection.from_pretrained("ultralytics/yolov5s")  # You may want to customize this model
-    return 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 windmill model: {str(e)}")