feat: display pretty json objects

app.py

@@ -94,13 +94,16 @@ 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)
+    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.write(f'\nRaw output a:\n{parsed_receipt_info_a}')
-    st.markdown(f'\nRaw output b:\n{parsed_receipt_info_b}')
+    parsed_receipt_info_b, _ = run_prediction(image)
+    st.text(f'\nRaw output a:')
+    st.json(parsed_receipt_info_a)
+    st.text(f'\nRaw output b:')
+    st.json(parsed_receipt_info_b)
 else:
     st.text(f'parsing 🧾...')
-    parsed_receipt_info = run_prediction(image)
-    st.write(f'\nRaw output:\n{parsed_receipt_info}')
+    parsed_receipt_info, _ = run_prediction(image)
+    st.text(f'\nRaw output')
+    st.json(parsed_receipt_info)