update dataframes

app.py

@@ -76,7 +76,7 @@ def draw_boxes(image, bounds, color='red', width=4):
     draw = ImageDraw.Draw(image)
     for i, bound in enumerate(bounds):
         p0, p1, p2, p3 = bound
-        draw.text(p0, str(i+1), fill=color, align='center')
+        draw.text((p0[0]+5, p0[1]+5), str(i+1), fill=color, align='center')
         draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
     return image
 
@@ -183,7 +183,7 @@ def ocr(img):
             result, scores = eval_step(task, generator, models, sample)
         ocr_result.append([str(i+1), result[0]['ocr'].replace(' ', '')])
 
-    result = pd.DataFrame(ocr_result)
+    result = pd.DataFrame(ocr_result, columns=['Box ID', 'Text'])
 
     return out_img, result
 
@@ -199,7 +199,7 @@ examples = [['shupai.png'], ['chinese.jpg'], ['gaidao.jpeg'], ['qiaodaima.png'],
             ['benpao.jpeg'], ['wanli.png'], ['xsd.jpg']]
 io = gr.Interface(fn=ocr, inputs=gr.inputs.Image(type='filepath', label='Image'),
                   outputs=[gr.outputs.Image(type='pil', label='Image'),
-                           gr.outputs.Dataframe(headers=['Box ID', 'Text'], type='pandas', label='OCR Results')],
+                           gr.outputs.Dataframe(type='pandas', label='OCR Results')],
                   title=title, description=description, article=article, examples=examples)
 io.launch()