debu das commited on
Commit
400d111
1 Parent(s): 1fc9f5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -432,7 +432,9 @@ class TableExtractionPipeline():
432
  '''
433
  image = Image.open(image_path).convert("RGB")
434
  model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=TD_THRESHOLD)
 
435
  st.image(image, caption='input')
 
436
  if bboxes_scaled.nelement() == 0:
437
  print('No table found in the pdf-page image'+image_path.split('/')[-1])
438
  return ''
 
432
  '''
433
  image = Image.open(image_path).convert("RGB")
434
  model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=TD_THRESHOLD)
435
+ st.text(bboxes_scaled)
436
  st.image(image, caption='input')
437
+ st.text(bboxes_scaled.nelement())
438
  if bboxes_scaled.nelement() == 0:
439
  print('No table found in the pdf-page image'+image_path.split('/')[-1])
440
  return ''