Spaces:
Runtime error
Runtime error
Commit
•
1f0478a
1
Parent(s):
30e33e1
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from transformers import LayoutLMForSequenceClassification
|
|
5 |
from preprocess import apply_ocr,encode_example
|
6 |
|
7 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
8 |
-
pytesseract.pytesseract.tesseract_cmd = r"C:\\Program Files\\Tesseract-OCR\\tesseract.exe"
|
9 |
model = LayoutLMForSequenceClassification.from_pretrained("models/document_model")
|
10 |
model.to(device)
|
11 |
classes=['questionnaire', 'memo', 'budget', 'file_folder', 'specification', 'invoice', 'resume',
|
|
|
5 |
from preprocess import apply_ocr,encode_example
|
6 |
|
7 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
8 |
+
# pytesseract.pytesseract.tesseract_cmd = r"C:\\Program Files\\Tesseract-OCR\\tesseract.exe"
|
9 |
model = LayoutLMForSequenceClassification.from_pretrained("models/document_model")
|
10 |
model.to(device)
|
11 |
classes=['questionnaire', 'memo', 'budget', 'file_folder', 'specification', 'invoice', 'resume',
|