fedihch commited on
Commit
a3b0dbe
1 Parent(s): 459740c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -9
README.md CHANGED
@@ -27,12 +27,4 @@ logits = outputs.logits
27
  predicted_class_idx = logits.argmax(-1).item()
28
  id2label = {0: "invoice", 1: "receipt"}
29
  print(id2label[predicted_class_idx])
30
- ```
31
-
32
- ---
33
- language:
34
- - code
35
- license: "other"
36
- datasets:
37
- - Custom dataset
38
- ---
 
27
  predicted_class_idx = logits.argmax(-1).item()
28
  id2label = {0: "invoice", 1: "receipt"}
29
  print(id2label[predicted_class_idx])
30
+ ```