Update app.py

app.py

@@ -15,8 +15,9 @@ import cv2
 import numpy as np
 # from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 # from cv2 import dnn_superres
-from transformers import DetrFeatureExtractor
-from transformers import DetrForObjectDetection
+#from transformers import DetrFeatureExtractor
+#from transformers import DetrForObjectDetection
+from transformers import TableTransformerForObjectDetection
 import torch
 import asyncio
 # pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
@@ -140,7 +141,7 @@ def table_detector(image, THRESHOLD_PROBA):
     feature_extractor = DetrFeatureExtractor(do_resize=True, size=800, max_size=800)
     encoding = feature_extractor(image, return_tensors="pt")
 
-    model = DetrForObjectDetection.from_pretrained("microsoft/table-transformer-detection")
+    model = TableTransformerForObjectDetection.from_pretrained("microsoft/table-transformer-detection")
 
     with torch.no_grad():
         outputs = model(**encoding)
@@ -163,7 +164,7 @@ def table_struct_recog(image, THRESHOLD_PROBA):
     feature_extractor = DetrFeatureExtractor(do_resize=True, size=1000, max_size=1000)
     encoding = feature_extractor(image, return_tensors="pt")
 
-    model = DetrForObjectDetection.from_pretrained("microsoft/table-transformer-structure-recognition")
+    model = TableTransformerForObjectDetection.from_pretrained("microsoft/table-transformer-structure-recognition")
     with torch.no_grad():
         outputs = model(**encoding)