Spaces:

H-TR
/

Donut_Herbarium_HTR

Runtime error

Jac-Zac commited on Aug 19, 2023

Commit

aa01645

1 Parent(s): 6fdbc56

update to new model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -150,12 +150,12 @@ if st.button("Parse sample! 🐍"):
         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"],
             )

         elif information == "Mid res new 0 epoch":
             processor = DonutProcessor.from_pretrained(
+                "Jac-Zac/thesis_donut",
                 #revision="4d64fa9a156908aa3df0e0e39463d401528a15c9",
                 use_auth_token=os.environ["TOKEN"],
             )
             pretrained_model = VisionEncoderDecoderModel.from_pretrained(
+                "Jac-Zac/thesis_donut",
                 #revision="4d64fa9a156908aa3df0e0e39463d401528a15c9",
                 use_auth_token=os.environ["TOKEN"],
             )