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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
handlers/messages_routing.py
View File

@ -112,7 +112,6 @@ async def messages_routing(msg, state):
text += '\n'
normalized_text = normalize(text)
logger.info(f'noramalized: {normalized_text}')
toxic_score = detector(normalized_text)
toxic_perc = math.floor(toxic_score * 100)
if toxic_score < 0.9 and text != text.replace(' ', ''):

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}')