File size: 174 Bytes
a204ca3
2170c2c
 
a204ca3
2170c2c
1
2
3
4
5
6
from transformers import AutoModelForObjectDetection

def load_solar_model():
    model = AutoModelForObjectDetection.from_pretrained("ultralytics/yolov5s")
    return model