Spaces:
Runtime error
Runtime error
chore: new messages as info
Browse files
app.py
CHANGED
@@ -87,24 +87,24 @@ with st.sidebar:
|
|
87 |
if img_file_buffer:
|
88 |
# To read image file buffer as a PIL Image:
|
89 |
photo = Image.open(img_file_buffer)
|
90 |
-
|
91 |
|
92 |
st.text(f'{information} mode is ON!\nTarget 🧾: {receipt}') # \n(opening image @:./img/receipt-{receipt}.png)')
|
93 |
|
94 |
-
if photo:
|
95 |
-
image = photo
|
96 |
-
print("photo loaded to image")
|
97 |
-
elif image_upload:
|
98 |
-
image = image_upload
|
99 |
-
else:
|
100 |
-
image = Image.open(f"./img/receipt-{receipt}.jpg")
|
101 |
-
|
102 |
col1, col2 = st.columns(2)
|
103 |
|
104 |
-
with
|
105 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
106 |
|
107 |
-
st.
|
108 |
|
109 |
if information == 'Receipt Summary':
|
110 |
processor = DonutProcessor.from_pretrained("unstructuredio/donut-base-sroie")
|
@@ -130,7 +130,7 @@ else:
|
|
130 |
|
131 |
with col2:
|
132 |
if information == 'Extract all':
|
133 |
-
st.
|
134 |
pretrained_model, processor, task_prompt = pretrained_model_a, processor_a, f"<s>"
|
135 |
pretrained_model.to(device)
|
136 |
parsed_receipt_info_a, _ = run_prediction(image)
|
@@ -142,7 +142,7 @@ with col2:
|
|
142 |
st.text(f'\nReceipt Menu Details:')
|
143 |
st.json(parsed_receipt_info_b)
|
144 |
else:
|
145 |
-
st.
|
146 |
parsed_receipt_info, _ = run_prediction(image)
|
147 |
st.text(f'\n{information}')
|
148 |
st.json(parsed_receipt_info)
|
|
|
87 |
if img_file_buffer:
|
88 |
# To read image file buffer as a PIL Image:
|
89 |
photo = Image.open(img_file_buffer)
|
90 |
+
st.info("picture taken!")
|
91 |
|
92 |
st.text(f'{information} mode is ON!\nTarget 🧾: {receipt}') # \n(opening image @:./img/receipt-{receipt}.png)')
|
93 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
94 |
col1, col2 = st.columns(2)
|
95 |
|
96 |
+
with st.spinner('Loading image...'):
|
97 |
+
if photo:
|
98 |
+
image = photo
|
99 |
+
st.info("photo loaded to image")
|
100 |
+
elif image_upload:
|
101 |
+
image = image_upload
|
102 |
+
else:
|
103 |
+
image = Image.open(f"./img/receipt-{receipt}.jpg")
|
104 |
+
with col1:
|
105 |
+
st.image(image, caption='Your target receipt')
|
106 |
|
107 |
+
st.info(f'baking the 🍩s...')
|
108 |
|
109 |
if information == 'Receipt Summary':
|
110 |
processor = DonutProcessor.from_pretrained("unstructuredio/donut-base-sroie")
|
|
|
130 |
|
131 |
with col2:
|
132 |
if information == 'Extract all':
|
133 |
+
st.info(f'parsing 🧾 (extracting all)...')
|
134 |
pretrained_model, processor, task_prompt = pretrained_model_a, processor_a, f"<s>"
|
135 |
pretrained_model.to(device)
|
136 |
parsed_receipt_info_a, _ = run_prediction(image)
|
|
|
142 |
st.text(f'\nReceipt Menu Details:')
|
143 |
st.json(parsed_receipt_info_b)
|
144 |
else:
|
145 |
+
st.info(f'parsing 🧾...')
|
146 |
parsed_receipt_info, _ = run_prediction(image)
|
147 |
st.text(f'\n{information}')
|
148 |
st.json(parsed_receipt_info)
|