laverdes commited on
Commit
4bf6412
·
1 Parent(s): 5279e45

feat: widgets to sidebar and message update

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -13,4 +13,15 @@ def demo_process(input_img):
13
 
14
  task_prompt = f"<s>"
15
 
16
- st.text('This model is trained with receipt images -> SROIE dataset.')
 
 
 
 
 
 
 
 
 
 
 
 
13
 
14
  task_prompt = f"<s>"
15
 
16
+ st.text('''
17
+ This is OCR-free Document Understanding Transformer nicknamed 🍩. It was fine-tuned with 1000 receipt images -> SROIE dataset.
18
+ The original 🍩 implementation can be found on: https://github.com/clovaai/donut
19
+ ''')
20
+
21
+ with st.sidebar:
22
+ information = st.radio(
23
+ "What information inside the are you interested in?",
24
+ ('Receipt Summary', 'Receipt Menu Details', 'Extract all!'))
25
+ receipt = st.selectbox('Pick one receipt', ['1', '2', '3', '4', '5', '6'])
26
+
27
+ st.text(f'{information} mode is ON!\nTarget receipt: {receipt}')