pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: judge not empty before delete (#10099)

Browse Source 
### What problem does this PR solve?

judge not empty before delete session.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Lynn
2025-09-15 17:49:52 +08:00
committed by GitHub
parent c29c395390
commit 341a7b1473
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
deepdoc/vision/ocr.py
View File

@ -351,7 +351,8 @@ class TextRecognizer:
def close(self):
# close session and release manually
logging.info('Close TextRecognizer.')
del self.predictor
if hasattr(self, "predictor"):
del self.predictor
gc.collect()
def __call__(self, img_list):
@ -490,7 +491,8 @@ class TextDetector:
def close(self):
logging.info("Close TextDetector.")
del self.predictor
if hasattr(self, "predictor"):
del self.predictor
gc.collect()
def __call__(self, img):

3
deepdoc/vision/recognizer.py
View File

@ -408,7 +408,8 @@ class Recognizer:
def close(self):
logging.info("Close recognizer.")
del self.ort_sess
if hasattr(self, "ort_sess"):
del self.ort_sess
gc.collect()
def __call__(self, image_list, thr=0.7, batch_size=16):