Spaces:
Runtime error
Runtime error
chore: update text messages and add commented code for file_upload
Browse files
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'\
|
102 |
st.json(parsed_receipt_info_a)
|
103 |
-
st.text(f'\
|
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'\
|
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)
|