From 56a7c0d839e6a52d42056d9c54aef63587af97d9 Mon Sep 17 00:00:00 2001
From: "aboutibm@163.com" <7p=e763wN3A6k+[C>
Date: Sat, 7 Sep 2024 15:20:53 +0800
Subject: [PATCH] model switch 'use_gpu' use same value from --gpu

---
 PPOCRLabel.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/PPOCRLabel.py b/PPOCRLabel.py
index 1d39a0b..0f4919a 100644
--- 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,7 +3182,7 @@ 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"]:
@@ -3189,7 +3190,7 @@ def modelChoose(self):
                     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,