laverdes commited on
Commit
f9ce3f3
·
1 Parent(s): 71bf396

chore: update text messages and add commented code for file_upload

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -61,6 +61,13 @@ with st.sidebar:
61
  ('Receipt Summary', 'Receipt Menu Details', 'Extract all!'))
62
  receipt = st.selectbox('Pick one 🧾', ['1', '2', '3', '4', '5', '6'], index=5)
63
 
 
 
 
 
 
 
 
64
  st.text(f'{information} mode is ON!\nTarget 🧾: {receipt}\n(opening image @:./img/receipt-{receipt}.png)')
65
 
66
  image = Image.open(f"./img/receipt-{receipt}.jpg")
@@ -98,12 +105,12 @@ if information == 'Extract all!':
98
  pretrained_model, processor, task_prompt = pretrained_model_b, processor_b, f"<s_cord-v2>"
99
  pretrained_model.to(device)
100
  parsed_receipt_info_b, _ = run_prediction(image)
101
- st.text(f'\nRaw output a:')
102
  st.json(parsed_receipt_info_a)
103
- st.text(f'\nRaw output b:')
104
  st.json(parsed_receipt_info_b)
105
  else:
106
  st.text(f'parsing 🧾...')
107
  parsed_receipt_info, _ = run_prediction(image)
108
- st.text(f'\nRaw output')
109
  st.json(parsed_receipt_info)
 
61
  ('Receipt Summary', 'Receipt Menu Details', 'Extract all!'))
62
  receipt = st.selectbox('Pick one 🧾', ['1', '2', '3', '4', '5', '6'], index=5)
63
 
64
+ # file upload
65
+ # uploaded_file = st.file_uploader("Choose a file")
66
+ # if uploaded_file is not None:
67
+ ## To read file as bytes:
68
+ # bytes_data = uploaded_file.getvalue()
69
+ # st.write(bytes_data)
70
+
71
  st.text(f'{information} mode is ON!\nTarget 🧾: {receipt}\n(opening image @:./img/receipt-{receipt}.png)')
72
 
73
  image = Image.open(f"./img/receipt-{receipt}.jpg")
 
105
  pretrained_model, processor, task_prompt = pretrained_model_b, processor_b, f"<s_cord-v2>"
106
  pretrained_model.to(device)
107
  parsed_receipt_info_b, _ = run_prediction(image)
108
+ st.text(f'\nReceipt Summary:')
109
  st.json(parsed_receipt_info_a)
110
+ st.text(f'\nReceipt Menu Details:')
111
  st.json(parsed_receipt_info_b)
112
  else:
113
  st.text(f'parsing 🧾...')
114
  parsed_receipt_info, _ = run_prediction(image)
115
+ st.text(f'\n{information}')
116
  st.json(parsed_receipt_info)