Spaces:

OFA-Sys
/

OFA-OCR

Runtime error

JustinLin610 commited on Nov 17, 2022

Commit

7961dd8

1 Parent(s): 97a6231

update dataframes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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).iloc[:, 1:]
     return out_img, result

             result, scores = eval_step(task, generator, models, sample)
         ocr_result.append([str(i+1), result[0]['ocr'].replace(' ', '')])
+    result = pd.DataFrame(ocr_result)
     return out_img, result