File size: 186 Bytes
7f4d7c9 |
1 2 3 4 5 6 7 |
from ultralytics import YOLO
fridgeModel = YOLO('AI_Model/Refrigerator.pt').cuda()
drinksModel = YOLO('AI_Model/softDrinks.pt').cuda()
fridgeModel.to(device=0)
drinksModel.to(device=0)
|