Update app.py

app.py

@@ -77,9 +77,9 @@ Experimental OCR-free Document Understanding Vision Transformer, fine-tuned with
 with st.sidebar:
     information = st.radio(
         "Choose one predictor:",
-        ("Low Res (1200 * 900) 5 epochs", "Mid res (1600 * 1200) 10 epochs", "Mid res (1600 * 1200) 14 epochs"),
+        ("Low Res (1200 * 900) 5 epochs", "Mid res (1600 * 1200) 10 epochs", "Mid res (1600 * 1200) 14 epochs"), "Mid res new 0 epoch"
     )
-    image_choice = st.selectbox("Pick one 📑", ["1", "2", "3"], index=0)
+    image_choice = st.selectbox("Pick one 📑", ["1", "2", "3","4"], index=0)
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
 
 st.text(
@@ -148,6 +148,17 @@ if st.button("Parse sample! 🐍"):
                 use_auth_token=os.environ["TOKEN"],
             )
 
+        elif information == "Mid res new 0 epoch":
+            processor = DonutProcessor.from_pretrained(
+                "Jac-Zac/thesis_test_donut",
+                revision="4d64fa9a156908aa3df0e0e39463d401528a15c9",
+                use_auth_token=os.environ["TOKEN"],
+            )
+            pretrained_model = VisionEncoderDecoderModel.from_pretrained(
+                "Jac-Zac/thesis_test_donut",
+                revision="4d64fa9a156908aa3df0e0e39463d401528a15c9",
+                use_auth_token=os.environ["TOKEN"],
+            )
 
         # this is the same for both models
         task_prompt = f"<s_herbarium>"