aho/welcomecenterbot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ocr-stav

Browse Source
This commit is contained in:
Untone
2024-09-29 13:24:49 +03:00
parent 335e65eaaa
commit a54d7cac39
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
nlp/ocr.py
View File

@@ -11,11 +11,12 @@ def ocr_recognize(file_path):
# Use EasyOCR to detect text in the photo
results = reader.readtext(file_path)
result = results[-1]
[_coords, ocr_text, ocr_accuracy] = result
logger.debug("OCR Result: %s", ocr_text)
if ocr_accuracy.item() > 0.5:
sum_text += " " + ocr_text
if isinstance(results, list):
result = results[-1]
[_coords, ocr_text, ocr_accuracy] = result
logger.debug("OCR Result: %s", ocr_text)
if ocr_accuracy.item() > 0.5:
sum_text += " " + ocr_text
logger.debug(f'Recognized Text: {sum_text}')