aiyubali
/

sofDrinks_AI

Model card Files Files and versions

sofDrinks_AI / Data /model.py

aiyubali's picture

server updated

7f4d7c9 over 1 year ago

history blame contribute delete

186 Bytes

	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)