patharanor commited on
Commit
402eb6e
·
1 Parent(s): 66fd3cf

feat: install tesseract-ocr in container image

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -5,6 +5,9 @@ import os
5
 
6
  layoutlm = None
7
 
 
 
 
8
  def auth(username, password):
9
  u = os.environ.get('USERNAME')
10
  p = os.environ.get('PASSWORD')
@@ -14,6 +17,9 @@ def inference(img) -> pd.DataFrame:
14
  return layoutlm.inference(img)
15
 
16
  if __name__ == "__main__":
 
 
 
17
  try:
18
  layoutlm = LayoutLM()
19
  layoutlm.set_model(layoutlm.default_model)
 
5
 
6
  layoutlm = None
7
 
8
+ def install_tesseract():
9
+ os.system('sudo apt-get install -y tesseract-ocr')
10
+
11
  def auth(username, password):
12
  u = os.environ.get('USERNAME')
13
  p = os.environ.get('PASSWORD')
 
17
  return layoutlm.inference(img)
18
 
19
  if __name__ == "__main__":
20
+
21
+ install_tesseract()
22
+
23
  try:
24
  layoutlm = LayoutLM()
25
  layoutlm.set_model(layoutlm.default_model)