rakib72642's picture
updated new sku name for version 12
22677f7
raw
history blame contribute delete
197 Bytes
from ultralytics import YOLO
uddoktaModel = YOLO('AI_Model/uddokta_v11.2.pt').cuda()
marchentModel = YOLO('AI_Model/marchent_v12.2.pt').cuda()
uddoktaModel.to(device=0)
marchentModel.to(device=0)