Update app.py

app.py

@@ -4,18 +4,14 @@ import os
 import TDTSR
 import pytesseract
 from pytesseract import Output
-import postprocess as pp
 import pandas as pd
 import matplotlib.pyplot as plt
 import cv2
 import numpy as np
-from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 from cv2 import dnn_superres
 
 pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
 
-
-
 st.set_option('deprecation.showPyplotGlobalUse', False)
 st.set_page_config(layout='wide')
 st.title("Table Detection and Table Structure Recognition")
@@ -32,19 +28,6 @@ def cv_to_PIL(cv_img):
 def pytess(cell_pil_img):
     return ' '.join(pytesseract.image_to_data(cell_pil_img, output_type=Output.DICT, config='preserve_interword_spaces')['text']).strip()
 
-def TrOCR(cell_pil_img):
-
-    processor = TrOCRProcessor.from_pretrained("SalML/trocr-base-printed")
-    model = VisionEncoderDecoderModel.from_pretrained("SalML/trocr-base-printed")
-    pixel_values = processor(images=cell_pil_img, return_tensors="pt").pixel_values
-
-    generated_ids = model.generate(pixel_values)
-    generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-
-    return generated_text
-
-
-
 def super_res(pil_img):
     # requires opencv-contrib-python installed without the opencv-python
     sr = dnn_superres.DnnSuperResImpl_create()