Spaces:
Runtime error
Runtime error
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"}
|