From f2f52467fc0b881ed9bd4ef33bbd3c9f47c55aac Mon Sep 17 00:00:00 2001 From: anjus Date: Wed, 7 Aug 2024 14:19:10 -0400 Subject: [PATCH] code cleanup and ruff fix --- app_doctr.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/app_doctr.py b/app_doctr.py index 96202d4..d7b1c7f 100644 --- a/app_doctr.py +++ b/app_doctr.py @@ -15,8 +15,6 @@ from doctr.models import ocr_predictor from img2table.document import Image from img2table.ocr import DocTR -import streamlit as st - # Hardcoded period types and formatting, probably won't update but can get them through API PERIOD_TYPES = { @@ -304,7 +302,7 @@ def clean_up(table_dfs): for col in range(table.shape[1]): cell_value = table.iloc[row][col] print(cell_value) - if cell_value == None: + if cell_value is None or cell_value=="None": table.iloc[row][col] = "" print(table.iloc[row][col]) print(table_dfs)