Update README.md

README.md

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