from ultralytics import YOLO

model_init= YOLO('model/pepsodent.pt').cuda()

model = model_init.to(device=0)