new: added dockerfile and app

Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.9
+
+WORKDIR /code
+
+RUN pip install --upgrade pip
+RUN pip install torch torchvision
+RUN pip install -U openmim
+RUN mim install mmengine
+RUN mim install mmcv
+RUN mim install mmdet
+
+RUN git clone https://github.com/open-mmlab/mmocr.git
+RUN cd mmocr
+RUN pip install -v -e .
+
+COPY . .
+
+CMD ["python", "app.py"]

app.py

@@ -1,7 +1,5 @@
-import streamlit as st
-from datasets import load_dataset
+from mmocr.apis import MMOCRInferencer
 
-dataset = load_dataset("katanaml-org/invoices-donut-data-v1")
-st.write(len(dataset))
-st.write(dir(dataset))
-st.write(type(dataset))
+infer = MMOCRInferencer(rec='svtr-small')
+result = infer('demo/demo_text_recog.jpg', save_vis=True, return_vis=True)
+print(result['predictions'])