Spaces:

bigodel
/

hydra-classifier

Sleeping

João Pedro commited on Jan 26

Commit

a3ab611

1 Parent(s): d73399c

typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ label2id = {v: k for k, v in id2label.items()}
 processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
 model = LayoutLMv3ForSequenceClassification.from_pretrained(
     "microsoft/layoutlmv3-base",
-    num_classes=len(labels),
     id2label=id2label,
     label2id=label2id,
 )

 processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
 model = LayoutLMv3ForSequenceClassification.from_pretrained(
     "microsoft/layoutlmv3-base",
+    num_labels=len(labels),
     id2label=id2label,
     label2id=label2id,
 )