debu das commited on
Commit
fb87466
1 Parent(s): 5f12024

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py CHANGED
@@ -20,6 +20,7 @@ from transformers import DetrFeatureExtractor
20
  from transformers import TableTransformerForObjectDetection
21
  import torch
22
  import asyncio
 
23
  # pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
24
 
25
 
@@ -41,6 +42,19 @@ async def pytess(cell_pil_img):
41
  print("pytess_output######################################")
42
  print(pytess_output)
43
  print("pytess_output@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
 
 
 
 
 
 
 
 
 
 
 
 
 
44
  return pytess_output
45
 
46
  # def super_res(pil_img):
 
20
  from transformers import TableTransformerForObjectDetection
21
  import torch
22
  import asyncio
23
+ import paddlehub as hub
24
  # pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
25
 
26
 
 
42
  print("pytess_output######################################")
43
  print(pytess_output)
44
  print("pytess_output@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
45
+ print("paddleocr$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$")
46
+ ###paddleocr
47
+ paddle_output=' '
48
+ cell_cv_img=PIL_to_cv(cell_pil_img)
49
+ ocr = PaddleOCR(use_angle_cls=True, lang=lang) # need to run only once to download and load model into memory
50
+ result = ocr.ocr(cell_cv_img, cls=True)
51
+ for idx in range(len(result)):
52
+ res = result[idx]
53
+ for line in res:
54
+ print(line)
55
+ paddle_output=paddle_output.join(line)
56
+ print(paddle_output)
57
+ print("paddleocr******************************************")
58
  return pytess_output
59
 
60
  # def super_res(pil_img):