Make sure download stats work on Hugging Face, improve metadata - #6

Thank you very much for your suggestions! If there are updates I will let you know immediately.

Hi @NielsRogge

Thanks for your valuable suggestions. We've now:

• Used PyTorchModelHubMixin for our model, enabling from_pretrained and push_to_hub.
• Ensured download stats are working.
• Created a Gradio demo: https://huggingface.co/spaces/opendatalab/DocLayout-YOLO
• Linked our datasets to the models and paper: https://huggingface.co/papers/2410.12628

Your assistance is greatly appreciated!

Best,
Bin

Thanks, looking great!

Note that for download stats to work, you will need to run the code snippet above for each of the model repos (basically overwriting them). As the class now inherits from PyTorchModelHubMixin, a config.json along with safetensors weights will be pushed.

Thanks, looking great!

Note that for download stats to work, you will need to run the code snippet above for each of the model repos (basically overwriting them). As the class now inherits from PyTorchModelHubMixin, a config.json along with safetensors weights will be pushed.

Noted, thanks for your contribution again!

Hi @JulioZhao97 are you planning to run the script above?

Cause I see downloads aren't working yet

@NielsRogge Hello! I tested the script above at the time you submit the PR and I find

model = YOLOv10.from_pretrained("juliozhao/DocLayout-YOLO-D4LA-from_scratch")

requires yolov10n.pt, and the model weights I upload to huggingface reposity have not been re-uploaded. But other in the script above is fine. I will see to it in these two days.

@NielsRogge Hello! I tried model.push_to_hub but no model weights found in the repository, could you please tell me where is wrong?

Follow is the target repository:

Hi @JulioZhao97 I just tried, it works for me: https://huggingface.co/nielsr/DocLayout-YOLO-D4LA-from_scratch/tree/main

Colab notebook: https://colab.research.google.com/drive/1N8Rfdu875lu5VCKG4uRUcM-Gi0o6qHom?usp=sharing.

Perhaps you might need to update your huggingface_hub version

Hey @JulioZhao97 how did you solve the pb to push to the hub ? Do you need to download the repo to do so as in Niels example ?

@agombert which error are you encountering?

@NielsRogge thanks for the quick answer

model = YOLOv10.from_pretrained("juliozhao/DocLayout-YOLO-D4LA-from_scratch")
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[91], [line 1](vscode-notebook-cell:?execution_count=91&line=1)
----> [1](vscode-notebook-cell:?execution_count=91&line=1) model = YOLOv10.from_pretrained("juliozhao/DocLayout-YOLO-D4LA-from_scratch")

AttributeError: type object 'YOLOv10' has no attribute 'from_pretrained'

It happens when I install doclayout-yolo = "^0.0.2.