Detect Language to lowercase

JamOnOff · Nov 28, 2023 · c104121 · c104121
1 parent 57a4b83
commit c104121
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/OCRController.py b/OCRController.py
@@ -25,12 +25,12 @@ def __initialize(self, detectLanguage):
             self.__device = 'cuda'
         else:
             self.__device = 'cpu'
-        
+
         # Inicializa EasyOCR con el dispositivo apropiado
-        if type(detectLanguage) is list:
-            self.__reader = easyocr.Reader(detectLanguage, model_storage_directory=None, gpu=self.__device)
+        if isinstance(detectLanguage, list):
+            self.__reader = easyocr.Reader([L.lower() for L in detectLanguage], model_storage_directory=None, gpu=self.__device)
         else:
-            self.__reader = easyocr.Reader([detectLanguage], model_storage_directory=None, gpu=self.__device)
+            self.__reader = easyocr.Reader([detectLanguage.lower()], model_storage_directory=None, gpu=self.__device)
 
     def __cleanData(self, data):
             """

diff --git a/__pycache__/OCRController.cpython-310.pyc b/__pycache__/OCRController.cpython-310.pyc