fix: pretrained_model unknown in extract all mode, markdown test

app.py

@@ -89,16 +89,17 @@ else:
     pretrained_model_b = VisionEncoderDecoderModel.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
     
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    pretrained_model.to(device)
 
 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.write(f'\nRaw output a:\n{parsed_receipt_info_a}')
-    st.write(f'\nRaw output b:\n{parsed_receipt_info_b}')
+    st.markdown(f'\nRaw output b:\n{parsed_receipt_info_b}')
 else:
     st.text(f'parsing 🧾...')
     parsed_receipt_info = run_prediction(image)