| from huggingface_hub import HfApi, list_repo_files |
|
|
| import os |
| TOKEN = os.environ.get("HF_TOKEN", "") |
| api = HfApi(token=TOKEN) |
|
|
| repos = sorted(api.list_models(author="Shanmuk4622"), key=lambda x: x.id) |
| print(f"Total repos on HF: {len(repos)}\n") |
|
|
| missing_readme = [] |
| missing_weights = [] |
|
|
| for r in repos: |
| files = list(list_repo_files(r.id, token=TOKEN)) |
| has_readme = "README.md" in files |
| has_weights = any(f.endswith(".pth") for f in files) |
| has_csv = any(f.endswith(".csv") for f in files) |
| parts = [] |
| if has_readme: parts.append("README") |
| if has_weights: parts.append("weights") |
| if has_csv: parts.append("CSV") |
| tag = "OK" if (has_readme and has_weights) else "!!" |
| name = r.id.replace("Shanmuk4622/", "") |
| print(f" [{tag}] {name:<45} [{' + '.join(parts)}]") |
| if not has_readme: missing_readme.append(name) |
| if not has_weights: missing_weights.append(name) |
|
|
| print() |
| print(f"Missing README : {missing_readme or 'None'}") |
| print(f"Missing weights : {missing_weights or 'None'}") |
|
|
