Release TAVI-PVR prediction models and dataset on Hugging Face

[NielsRogge]

Hi Michele Cannito 🤗

I'm Niels and work as part of the open-source team at Hugging Face. I discovered your work on Arxiv and was wondering whether you would like to submit it to hf.co/papers to improve its discoverability. If you are one of the authors, you can submit it at https://huggingface.co/papers/submit.

The paper page lets people discuss about your paper and lets them find artifacts about it (your models or dataset for instance), you can also claim the paper as yours which will show up on your public profile at HF, add Github and project page URLs.

I noticed that the paper mentions the source code is available on GitHub at https://github.com/EIDOSLAB/tavi, though the link currently seems to be unavailable. Would you like to host the 3D models you've trained (the 3D DenseNet and ResNet variants) and potentially the dataset on Hugging Face? Hosting on the Hub will give your work more visibility and enable better discoverability within the medical imaging community.

If you're down, leaving a guide here. If it's a custom PyTorch model, you can use the PyTorchModelHubMixin class which adds from_pretrained and push_to_hub to the model which lets you to upload the model and people to download and use models right away.

Additionally, if you are able to share the CT dataset (or a subset/anonymized version), hosting it on https://huggingface.co/datasets would allow others to easily load it via:

from datasets import load_dataset
dataset = load_dataset("your-username/tavi-dataset")

Let me know if you're interested or need any guidance :)

Kind regards,

Niels