Updated model to use

app.py

@@ -26,7 +26,8 @@ from transformers import LayoutLMv3FeatureExtractor, LayoutLMv3TokenizerFast, La
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 TOKENIZER = "microsoft/layoutlmv3-base"
-MODEL_NAME = "fsommers/layoutlmv3-autofinance-classification-us-v01"
+#MODEL_NAME = "fsommers/layoutlmv3-autofinance-classification-us-v01"
+MODEL_NAME = "fsommers/layoutlmv3-FinDocs-sequence-classification-2-12-5-2934-0.0651"
 
 TESS_OPTIONS = "--psm 3" # Automatic page segmentation for Tesseract
 
@@ -137,4 +138,4 @@ if uploaded_file is not None:
     })
     fig = px.bar(df, x="Label", y="Probability")
     st.plotly_chart(fig, use_container_width=True)
-    
+