From fff961fdd92c0b7fa685952708bd72b7cc6f0d82 Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Mon, 7 Apr 2025 19:28:04 -0500
Subject: [PATCH] use device

---
 frigate/data_processing/common/license_plate/model.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/frigate/data_processing/common/license_plate/model.py b/frigate/data_processing/common/license_plate/model.py
index b22ab73a5f..25e7b2cafc 100644
--- a/frigate/data_processing/common/license_plate/model.py
+++ b/frigate/data_processing/common/license_plate/model.py
@@ -12,13 +12,13 @@ class LicensePlateModelRunner(DataProcessorModelRunner):
     def __init__(self, requestor, device: str = "CPU", model_size: str = "large"):
         super().__init__(requestor, device, model_size)
         self.detection_model = PaddleOCRDetection(
-            model_size=model_size, requestor=requestor, device="CPU"
+            model_size=model_size, requestor=requestor, device=device
         )
         self.classification_model = PaddleOCRClassification(
-            model_size=model_size, requestor=requestor, device="CPU"
+            model_size=model_size, requestor=requestor, device=device
         )
         self.recognition_model = PaddleOCRRecognition(
-            model_size=model_size, requestor=requestor, device="CPU"
+            model_size=model_size, requestor=requestor, device=device
         )
         self.yolov9_detection_model = LicensePlateDetector(
             model_size=model_size, requestor=requestor, device=device