fedihch
/

InvoiceReceiptClassifier

Image Classification

text-classification

Model card Files Files and versions

fedihch commited on Aug 5, 2022

Commit

a3b0dbe

·

1 Parent(s): 459740c

Update README.md

Files changed (1) hide show

README.md +1 -9

README.md CHANGED Viewed

@@ -27,12 +27,4 @@ logits = outputs.logits
 predicted_class_idx = logits.argmax(-1).item()
 id2label = {0: "invoice", 1: "receipt"}
 print(id2label[predicted_class_idx])
-```
----
-language:
-  - code
-license: "other"
-datasets:
-- Custom dataset
----

 predicted_class_idx = logits.argmax(-1).item()
 id2label = {0: "invoice", 1: "receipt"}
 print(id2label[predicted_class_idx])
+```