From 1b4dfa4449bb0c815d23a09ec8ce2078b2f0b797 Mon Sep 17 00:00:00 2001
From: Bigsk <i@ianxia.com>
Date: Fri, 27 Jan 2023 13:41:39 +0800
Subject: [PATCH] fixed a cuda config bug

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 9d6795e..eb29fa6 100644
--- a/main.py
+++ b/main.py
@@ -23,7 +23,7 @@ class number(object):
         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)