keldrenloy commited on
Commit
da66d52
·
1 Parent(s): c9bed60

update model in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/layoutlmv3cordfinetuned") #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
 
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