from ultralytics import YOLO OCR_model = YOLO("AI_Model/ArabicOCR.pt").cuda() OCR_model.to(device=0)