nagad_django_v13 / model /model.py
rakib72642's picture
build architecture
07d2322
raw
history blame contribute delete
209 Bytes
from ultralytics import YOLO
uddoktaModel = YOLO('model/AI_Model/uddokta_v13.1.pt').cuda()
marchentModel = YOLO('model/AI_Model/marchent_v13.1.pt').cuda()
uddoktaModel.to(device=0)
marchentModel.to(device=0)