EC2 Default User commited on
Commit
2c64a65
·
1 Parent(s): 7235802

change column name

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -194,8 +194,9 @@ def visualize_cells(image, table_structures, cells):
194
  draw.text((x1-20, y1), str(row_num)+'-'+str(col_num), (255,0,0), font=fontStyle)
195
  draw.text((x1, y1), text, (0,0,255), font=fontStyle)
196
 
197
- df = pd.DataFrame(data_rows[1:], columns=data_rows[0])
198
- return image, df, df.to_json(orient='records')
 
199
 
200
 
201
  def ocr(image):
 
194
  draw.text((x1-20, y1), str(row_num)+'-'+str(col_num), (255,0,0), font=fontStyle)
195
  draw.text((x1, y1), text, (0,0,255), font=fontStyle)
196
 
197
+ df = pd.DataFrame(data_rows)
198
+ df.columns = df.columns.astype(str)
199
+ return image, df, df.to_json()
200
 
201
 
202
  def ocr(image):