Spaces:

unstructuredio
/

unstructured-chipper-app

Running

ajimeno commited on Oct 4, 2023

Commit

43aafa0

1 Parent(s): a056c48

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ else:
 st.image(image, caption='Your target document')
 with st.spinner(f'Processing the document ...'):
-        pre_trained_model = "unstructuredio/chipper-fast-fine-tuning"
         processor = DonutProcessor.from_pretrained(pre_trained_model, token=os.environ['HF_TOKEN'])
         device = "cuda" if torch.cuda.is_available() else "cpu"

 st.image(image, caption='Your target document')
 with st.spinner(f'Processing the document ...'):
+        pre_trained_model = "unstructuredio/chipper-fast-fine-tuning-v2"
         processor = DonutProcessor.from_pretrained(pre_trained_model, token=os.environ['HF_TOKEN'])
         device = "cuda" if torch.cuda.is_available() else "cpu"