model switch 'use_gpu' use same value from --gpu

PFCCLab · Sep 7, 2024 · 56a7c0d · 56a7c0d
1 parent f90256a
commit 56a7c0d
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,