architsingh/InvoiceExtractor

1from datasets import load_dataset2import streamlit as st3from mmocr.apis import MMOCRInferencer4import mmcv5import matplotlib.pyplot as plt6 7st.write('Hello, *World!* :sunglasses:')8dataset = load_dataset("katanaml-org/invoices-donut-data-v1")9st.write(f"{len(dataset)}")10 11infer = MMOCRInferencer(det='dbnetpp', rec='svtr-small', kie='SDMGR')12result = infer('demo/demo_kie.jpeg', save_vis=True)13 14predicted_img = mmcv.imread('results/vis/demo_kie.jpg')15plt.figure(figsize=(18, 32))16plt.imshow(mmcv.bgr2rgb(predicted_img))17plt.show()18