Hugging Face's logo

Spaces:
fsommers
/
autofinance-us
Running

App Files Community
fsommers commited on
Commit
d185e95
•
1 Parent(s): 5fe63a3

Reverted model reference

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -26,8 +26,7 @@ from transformers import LayoutLMv3FeatureExtractor, LayoutLMv3TokenizerFast, La
26
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
27
 
28
  TOKENIZER = "microsoft/layoutlmv3-base"
29
- #MODEL_NAME = "fsommers/layoutlmv3-autofinance-classification-us-v01"
30
- MODEL_NAME = "fsommers/layoutlmv3-FinDocs-sequence-classification-2-12-5-2934-0.0651"
31
 
32
  TESS_OPTIONS = "--psm 3" # Automatic page segmentation for Tesseract
33
 
 
26
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
27
 
28
  TOKENIZER = "microsoft/layoutlmv3-base"
29
+ MODEL_NAME = "fsommers/layoutlmv3-autofinance-classification-us-v01"
 
30
 
31
  TESS_OPTIONS = "--psm 3" # Automatic page segmentation for Tesseract
32