feat: column separation

app.py

@@ -71,7 +71,11 @@ with st.sidebar:
 st.text(f'{information} mode is ON!\nTarget 🧾: {receipt}\n(opening image @:./img/receipt-{receipt}.png)')
 
 image = Image.open(f"./img/receipt-{receipt}.jpg")
-st.image(image, caption='Your target receipt')
+
+col1, col2 = st.columns(2)
+
+with col1:
+    st.image(image, caption='Your target receipt')
 
 st.text(f'baking the 🍩s...')
 
@@ -97,20 +101,21 @@ else:
     
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
-if information == 'Extract all!':
-    st.text(f'parsing 🧾 (extracting all)...')
-    pretrained_model, processor, task_prompt = pretrained_model_a, processor_a, f"<s>"
-    pretrained_model.to(device)
-    parsed_receipt_info_a, _ = run_prediction(image)
-    pretrained_model, processor, task_prompt = pretrained_model_b, processor_b, f"<s_cord-v2>"
-    pretrained_model.to(device)
-    parsed_receipt_info_b, _ = run_prediction(image)
-    st.text(f'\nReceipt Summary:')
-    st.json(parsed_receipt_info_a)
-    st.text(f'\nReceipt Menu Details:')
-    st.json(parsed_receipt_info_b)
-else:
-    st.text(f'parsing 🧾...')
-    parsed_receipt_info, _ = run_prediction(image)
-    st.text(f'\n{information}')
-    st.json(parsed_receipt_info)
+with col2:
+    if information == 'Extract all!':
+        st.text(f'parsing 🧾 (extracting all)...')
+        pretrained_model, processor, task_prompt = pretrained_model_a, processor_a, f"<s>"
+        pretrained_model.to(device)
+        parsed_receipt_info_a, _ = run_prediction(image)
+        pretrained_model, processor, task_prompt = pretrained_model_b, processor_b, f"<s_cord-v2>"
+        pretrained_model.to(device)
+        parsed_receipt_info_b, _ = run_prediction(image)
+        st.text(f'\nReceipt Summary:')
+        st.json(parsed_receipt_info_a)
+        st.text(f'\nReceipt Menu Details:')
+        st.json(parsed_receipt_info_b)
+    else:
+        st.text(f'parsing 🧾...')
+        parsed_receipt_info, _ = run_prediction(image)
+        st.text(f'\n{information}')
+        st.json(parsed_receipt_info)