bachpc commited on
Commit
a4d6532
1 Parent(s): c9a4627
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -291,13 +291,13 @@ def visualize_cells(pil_img, cells):
291
  for cell in cells:
292
  bbox = cell['bbox']
293
 
294
- if cell['column header']:
295
  facecolor = (1, 0, 0.45)
296
  edgecolor = (1, 0, 0.45)
297
  alpha = 0.3
298
  linewidth = 2
299
  hatch='//////'
300
- elif cell['projected row header']:
301
  facecolor = (0.95, 0.6, 0.1)
302
  edgecolor = (0.95, 0.6, 0.1)
303
  alpha = 0.3
 
291
  for cell in cells:
292
  bbox = cell['bbox']
293
 
294
+ if cell['header']:
295
  facecolor = (1, 0, 0.45)
296
  edgecolor = (1, 0, 0.45)
297
  alpha = 0.3
298
  linewidth = 2
299
  hatch='//////'
300
+ elif cell['subheader']:
301
  facecolor = (0.95, 0.6, 0.1)
302
  edgecolor = (0.95, 0.6, 0.1)
303
  alpha = 0.3