Sage commited on
Commit
c6dfe38
1 Parent(s): 3be8992
Files changed (2) hide show
  1. app.py +2 -2
  2. ocr_functions.py +1 -0
app.py CHANGED
@@ -69,8 +69,8 @@ def batch_document(content, document_type, context, progress = gr.Progress()):
69
  else:
70
  progress(0, desc="Starting")
71
  for x in progress.tqdm(content, desc="Processing"):
72
- retries = 1
73
- timeout = 1
74
  i = 0
75
  while True:
76
  try:
 
69
  else:
70
  progress(0, desc="Starting")
71
  for x in progress.tqdm(content, desc="Processing"):
72
+ retries = 3
73
+ timeout = 3
74
  i = 0
75
  while True:
76
  try:
ocr_functions.py CHANGED
@@ -50,6 +50,7 @@ def detect_document(content):
50
  )
51
  logging.info(table_content)
52
  name = content.name.split('\\')[-1]
 
53
  name = name.split('.')[0]
54
  return (pair_content, document_content, table_content, name)
55
 
 
50
  )
51
  logging.info(table_content)
52
  name = content.name.split('\\')[-1]
53
+ name = name.split("/")[-1]
54
  name = name.split('.')[0]
55
  return (pair_content, document_content, table_content, name)
56