From becb5eaf0815261c4f79a77f148dfe800c7e0403 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Sun, 7 Jul 2024 11:19:17 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=86ocr=E5=8A=A0=E8=BD=BD=E5=88=B0gpu1?=
 =?UTF-8?q?=E4=B8=8A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 photo_review/__init__.py |  2 +-
 test.py                  | 21 ---------------------
 2 files changed, 1 insertion(+), 22 deletions(-)
 delete mode 100644 test.py

diff --git a/photo_review/__init__.py b/photo_review/__init__.py
index 368334b..872bc2c 100644
--- a/photo_review/__init__.py
+++ b/photo_review/__init__.py
@@ -10,4 +10,4 @@ DISCHARGE_IE = Taskflow("information_extraction", schema=keys.DISCHARGE_RECORD_S
 COST_IE = Taskflow("information_extraction", schema=keys.COST_LIST_SCHEMA, model="uie-x-base", device_id=1,
                    task_path="config/model/cost_list_model", layout_analysis=False, precision='fp16')
 
-OCR = PaddleOCR(use_angle_cls=False, lang="ch", show_log=False)
+OCR = PaddleOCR(use_angle_cls=False, lang="ch", show_log=False, gpu_id=1)
diff --git a/test.py b/test.py
deleted file mode 100644
index 419668e..0000000
--- a/test.py
+++ /dev/null
@@ -1,21 +0,0 @@
-from paddlenlp import Taskflow
-
-SCHEMA = ["患者姓名", "入院日期", "出院日期", "费用总额", "个人现金支付", "个人账户支付", "自费金额", "医保类型",
-          "住院号", "医保结算单号码"]
-
-IE = Taskflow("information_extraction", schema=SCHEMA, model="uie-x-base",
-              task_path="config/model/settlement_list_model", layout_analysis=True)
-
-
-def main():
-    i = 0
-    try:
-        while 1:
-            IE({"doc": "visual_model_test/img/PH20240511000638_1_094306_1.jpg"})
-            i += 1
-    except Exception:
-        print(i)
-
-
-if __name__ == '__main__':
-    main()