Counting publicly available Heretic models and their downloads

from huggingface_hub import HfApi


def main():
    api = HfApi()

    count = 0
    total_downloads = 0
    total_downloads_last_30_days = 0

    tagged_models = api.list_models(
        filter="heretic",
        sort="downloads",
        expand=["downloads", "downloadsAllTime"],
    )

    tagged_model_ids = set()

    for model in tagged_models:
        count += 1
        total_downloads += model.downloads_all_time
        total_downloads_last_30_days += model.downloads
        tagged_model_ids.add(model.id)
        print(
            f"[TAGGED] https://huggingface.co/{model.id}: {model.downloads_all_time} downloads, {model.downloads} in the last 30 days"
        )

    named_models = api.list_models(
        search="heretic",
        sort="downloads",
        expand=["downloads", "downloadsAllTime"],
    )

    for model in named_models:
        if model.id in tagged_model_ids:
            continue
        count += 1
        total_downloads += model.downloads_all_time
        total_downloads_last_30_days += model.downloads
        print(
            f"[NAMED] https://huggingface.co/{model.id}: {model.downloads_all_time} downloads, {model.downloads} in the last 30 days"
        )

    print()
    print(
        f"{count} Heretic models with {total_downloads} total downloads, {total_downloads_last_30_days} in the last 30 days"
    )


if __name__ == "__main__":
    main()