Spaces:
Running
Running
File size: 948 Bytes
04a30fc c3ede35 15e7787 73834eb 137bc36 6d85f4d 137bc36 6d85f4d 04a30fc 73834eb 04a30fc 137bc36 04a30fc 137bc36 04a30fc 280d208 9f02f99 280d208 04a30fc |
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 34 35 36 37 38 39 40 |
from huggingface_hub import HfApi, RepoUrl
async def launch_substra_space(
hf_api: HfApi, repo_id: str,
hospital_a: int, hospital_b: int,
) -> RepoUrl:
repo_id = "owkin/trainer-" + repo_id
BASE_IMAGE = "ghcr.io/substra/substrafl:substra-hf-demo"
DOCKERFILE = f"""
FROM {BASE_IMAGE}
ENV SUBSTRA_ORG1_DISTR={hospital_a / 100}
ENV SUBSTRA_ORG2_DISTR={hospital_b / 100}
CMD ["bash", "docker-run.sh"]
"""
repo_url = hf_api.create_repo(
repo_id=repo_id,
repo_type="space",
space_sdk="docker",
)
hf_api.upload_file(
repo_id=repo_id,
path_or_fileobj=DOCKERFILE.encode(),
path_in_repo="Dockerfile",
repo_type="space",
)
hf_api.upload_file(
repo_id=repo_id,
path_or_fileobj="./substra_template/README.md",
path_in_repo="README.md",
repo_type="space",
)
return repo_url
|