confia-backend / init.py
elsamueldev's picture
fix: actualizar el script de inicializaci贸n para ejecutar main.py despu茅s de la descarga
21d4b07
raw
history blame contribute delete
614 Bytes
import os
import subprocess
from huggingface_hub import login, snapshot_download
HF_TOKEN = os.getenv("HF_TOKEN")
if HF_TOKEN is None:
raise RuntimeError("HF_TOKEN is not set")
CODE_REPO_ID = "elsamueldev/confia-backend-code"
login(HF_TOKEN)
snapshot_download(
repo_id=CODE_REPO_ID,
repo_type="dataset",
local_dir=".",
local_dir_use_symlinks=False
)
local_dir = subprocess.run(["ls"], stdout=subprocess.PIPE).stdout.decode("utf-8")
print(local_dir)
if "main.py" not in local_dir:
raise RuntimeError("main.py not found in the downloaded dataset")
subprocess.run(["python", "main.py"])