Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ import torch
|
|
5 |
|
6 |
# Load the fine-tuned model and processor
|
7 |
model_path = "quadranttechnologies/Receipt_Image_Analyzer"
|
8 |
-
model = LayoutLMv3ForTokenClassification.from_pretrained(
|
9 |
-
processor = LayoutLMv3Processor.from_pretrained(
|
10 |
|
11 |
# Define label mapping
|
12 |
id2label = {0: "company", 1: "date", 2: "address", 3: "total", 4: "other"}
|
|
|
5 |
|
6 |
# Load the fine-tuned model and processor
|
7 |
model_path = "quadranttechnologies/Receipt_Image_Analyzer"
|
8 |
+
model = LayoutLMv3ForTokenClassification.from_pretrained(quadranttechnologies/Receipt_Image_Analyzer)
|
9 |
+
processor = LayoutLMv3Processor.from_pretrained(quadranttechnologies/Receipt_Image_Analyzer)
|
10 |
|
11 |
# Define label mapping
|
12 |
id2label = {0: "company", 1: "date", 2: "address", 3: "total", 4: "other"}
|