Spaces:
Runtime error
Runtime error
Commit
·
da66d52
1
Parent(s):
c9bed60
update model in app.py
Browse files
app.py
CHANGED
@@ -80,7 +80,7 @@ def unnormalize_box(bbox, width, height):
|
|
80 |
]
|
81 |
|
82 |
def predict(image):
|
83 |
-
model = LayoutLMv3ForTokenClassification.from_pretrained("keldrenloy/
|
84 |
processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
|
85 |
label_list,id2label,label2id, num_labels = convert_l2n_n2l(dataset)
|
86 |
width, height = image.size
|
|
|
80 |
]
|
81 |
|
82 |
def predict(image):
|
83 |
+
model = LayoutLMv3ForTokenClassification.from_pretrained("keldrenloy/LayoutLMv3FineTunedwithCORDandSGReceipts") #add your model directory here
|
84 |
processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
|
85 |
label_list,id2label,label2id, num_labels = convert_l2n_n2l(dataset)
|
86 |
width, height = image.size
|