diff --git a/handlers/messages_routing.py b/handlers/messages_routing.py index a93c4a9..608d354 100644 --- a/handlers/messages_routing.py +++ b/handlers/messages_routing.py @@ -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(' ', ''): diff --git a/nlp/ocr.py b/nlp/ocr.py index bd287ad..00e1ccc 100644 --- a/nlp/ocr.py +++ b/nlp/ocr.py @@ -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}')