Dileep7729 commited on
Commit
86424d0
·
verified ·
1 Parent(s): 0320eb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(model_path)
9
- processor = LayoutLMv3Processor.from_pretrained(model_path)
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"}