model switch 'use_gpu' use the same value from --gpu (#68)

* model switch 'use_gpu' use the same value from --gpu * fix code style --------- Co-authored-by: [email protected] <7p=e763wN3A6k+[C> Co-authored-by: Wang Xin <[email protected]>
PFCCLab · Sep 7, 2024 · ff235d3 · ff235d3
1 parent f90256a
commit ff235d3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/PPOCRLabel.py b/PPOCRLabel.py
@@ -144,6 +144,7 @@ def __init__(
         self.settings.load()
         settings = self.settings
         self.lang = lang
+        self.gpu = gpu
 
         # Load string bundle for i18n
         if lang not in ["ch", "en"]:
@@ -3181,15 +3182,15 @@ def modelChoose(self):
                 use_angle_cls=True,
                 det=True,
                 cls=True,
-                use_gpu=False,
+                use_gpu=self.gpu,
                 lang=choose_lang,
             )
             if choose_lang in ["ch", "en"]:
                 if hasattr(self, "table_ocr"):
                     del self.table_ocr
                 self.table_ocr = PPStructure(
                     use_pdserving=False,
-                    use_gpu=False,
+                    use_gpu=self.gpu,
                     lang=choose_lang,
                     layout=False,
                     show_log=False,