| import sys | |
| from huggingface_hub import snapshot_download, upload_folder | |
| REPO = "EagleDev-Innovation/Modelos-EagleDev-Innovation" | |
| CARPETA = "modelos_entrenados" | |
| def bajar(): | |
| snapshot_download( | |
| repo_id=REPO, | |
| repo_type="model", | |
| allow_patterns=f"{CARPETA}/**", | |
| local_dir=".", | |
| ) | |
| print("Modelos descargados en", CARPETA) | |
| def subir(mensaje): | |
| upload_folder( | |
| repo_id=REPO, | |
| repo_type="model", | |
| folder_path=CARPETA, | |
| path_in_repo=CARPETA, | |
| commit_message=mensaje, | |
| ignore_patterns=[".locks/**", ".cache/**", "__pycache__/**", "*.tmp"], | |
| ) | |
| print("Cambios subidos") | |
| if __name__ == "__main__": | |
| accion = sys.argv[1] | |
| if accion == "bajar": | |
| bajar() | |
| elif accion == "subir": | |
| subir(sys.argv[2] if len(sys.argv) > 2 else "Actualizar modelos") | |