revert back to using pre-trained layoutlm

app.py

@@ -14,10 +14,26 @@ if not wandb_api_key:
 else:
     wandb.login(key=wandb_api_key)
 
+labels = [
+    'budget',
+    'email',
+    'form',
+    'handwritten',
+    'invoice',
+    'language',
+    'letter',
+    'memo',
+    'news article',
+    'questionnaire',
+    'resume',
+    'scientific publication',
+    'specification',
+]
+id2label = {i: label for i, label in enumerate(labels)}
+label2id = {v: k for k, v in id2label.items()}
+
 processor = LayoutLMv3Processor.from_pretrained("model/layoutlmv3/")
 model = LayoutLMv3ForSequenceClassification.from_pretrained("model/layoutlmv3/")
-id2label = model.config.id2label
-label2id = model.config.label2id
 
 st.title("Document Classification with LayoutLMv3")