Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -81,7 +81,7 @@ else:
|
|
| 81 |
st.image(image, caption='Your target document')
|
| 82 |
|
| 83 |
with st.spinner(f'Processing the document ...'):
|
| 84 |
-
pre_trained_model = "unstructuredio/chipper-fast-fine-tuning"
|
| 85 |
processor = DonutProcessor.from_pretrained(pre_trained_model, token=os.environ['HF_TOKEN'])
|
| 86 |
|
| 87 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 81 |
st.image(image, caption='Your target document')
|
| 82 |
|
| 83 |
with st.spinner(f'Processing the document ...'):
|
| 84 |
+
pre_trained_model = "unstructuredio/chipper-fast-fine-tuning-v2"
|
| 85 |
processor = DonutProcessor.from_pretrained(pre_trained_model, token=os.environ['HF_TOKEN'])
|
| 86 |
|
| 87 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|