Harsimran19 commited on
Commit
1f0478a
1 Parent(s): 30e33e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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',