nielsr HF staff commited on
Commit
5279027
1 Parent(s): 3a81b9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -141,7 +141,7 @@ def recognize_table(image):
141
  # postprocess to get individual elements
142
  id2label = structure_model.config.id2label
143
  id2label[len(structure_model.config.id2label)] = "no object"
144
- detected_tables = outputs_to_objects(outputs, image.size, id2label)
145
 
146
  # visualize cells on cropped table
147
  draw = ImageDraw.Draw(image)
 
141
  # postprocess to get individual elements
142
  id2label = structure_model.config.id2label
143
  id2label[len(structure_model.config.id2label)] = "no object"
144
+ cells = outputs_to_objects(outputs, image.size, id2label)
145
 
146
  # visualize cells on cropped table
147
  draw = ImageDraw.Draw(image)