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)