Update app.py
Browse files
app.py
CHANGED
@@ -21,8 +21,8 @@ except Exception as e:
|
|
21 |
# pytesseract.pytesseract.tesseract_cmd = r"C:\Program Files\Tesseract-OCR\tesseract.exe"
|
22 |
|
23 |
# Load the model and processor
|
24 |
-
processor = LayoutLMv3Processor.from_pretrained("quadranttechnologies/
|
25 |
-
model = LayoutLMv3ForTokenClassification.from_pretrained("quadranttechnologies/
|
26 |
model.eval()
|
27 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
28 |
model.to(device)
|
|
|
21 |
# pytesseract.pytesseract.tesseract_cmd = r"C:\Program Files\Tesseract-OCR\tesseract.exe"
|
22 |
|
23 |
# Load the model and processor
|
24 |
+
processor = LayoutLMv3Processor.from_pretrained("quadranttechnologies/Receipt_Image_Analyzer")
|
25 |
+
model = LayoutLMv3ForTokenClassification.from_pretrained("quadranttechnologies/Receipt_Image_Analyzer")
|
26 |
model.eval()
|
27 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
28 |
model.to(device)
|