Update substra_launcher.py

substra_launcher.py

@@ -7,8 +7,16 @@ async def launch_substra_space(
 ) -> 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} 
     
-    ENV_FILE = f"SUBSTRA_ORG1_DISTR={hospital_a / 100} \nSUBSTRA_ORG2_DISTR={hospital_b / 100}"
+    CMD ["bash", "docker-run.sh"]
+    """
 
     with open("substra_template/.env", 'w') as f:
         f.write(ENV_FILE)
@@ -19,10 +27,10 @@ async def launch_substra_space(
         space_sdk="docker",
     )
 
-    hf_api.upload_folder(
+    hf_api.upload_file(
         repo_id=repo_id,
+        path_or_fileobj=DOCKERFILE.encode(),
+        path_in_repo="Dockerfile",
         repo_type="space",
-        folder_path="./substra_template/"
     )
-
     return repo_url