Update README.md

README.md

@@ -1,3 +1,51 @@
----
-license: mit
----
+---
+license: mit
+base_model:
+- google/efficientnet-b0
+---
+
+
+# EfficientNet-B0 Document Image Classifier
+
+This is an image classification model based on **Google EfficientNet-B0**, fine-tuned to classify input images into one of the following 16 categories:
+
+1. **bar_chart**
+2. **bar_code**
+3. **chemistry_markush_structure**
+4. **chemistry_molecular_structure**
+5. **flow_chart**
+6. **icon**
+7. **line_chart**
+8. **logo**
+9. **map**
+10. **other**
+11. **pie_chart**
+12. **qr_code**
+13. **remote_sensing**
+14. **screenshot**
+15. **signature**
+16. **stamp**
+
+## Citation
+If you use this model in your work, please cite the following papers:
+
+```
+@article{Tan2019EfficientNetRM,
+  title={EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks},
+  author={Mingxing Tan and Quoc V. Le},
+  journal={ArXiv},
+  year={2019},
+  volume={abs/1905.11946}
+}
+
+@techreport{Docling,
+  author = {Deep Search Team},
+  month = {8},
+  title = {{Docling Technical Report}},
+  url={https://arxiv.org/abs/2408.09869},
+  eprint={2408.09869},
+  doi = "10.48550/arXiv.2408.09869",
+  version = {1.0.0},
+  year = {2024}
+}
+```