File size: 856 Bytes
a294091 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | 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")
|