fixed a cuda config bug

ghinknet · Jan 27, 2023 · 1b4dfa4 · 1b4dfa4
1 parent 6fed221
commit 1b4dfa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -23,7 +23,7 @@ def __init__(self, gpu=False, times=1, filter=0.8, ua="ARNRS", debug=False):
         self.debug = debug
 
         # Init detector and OCR
-        self.__detector = Detector(device="gpu" if gpu else "cpu")
+        self.__detector = Detector(device="cuda" if gpu else "cpu")
         self.__eocr = easyocr.Reader(['ch_sim', 'en'], gpu=self.gpu)
         self.__pocr = PaddleOCR(use_angle_cls=True, use_gpu=self.gpu, show_log=debug)