fix: actualizar el script de inicialización para ejecutar main.py después de la descarga

Dockerfile

@@ -17,8 +17,5 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 # copy repo files
 COPY --chown=user . /app
 
-# run initialization script that downloads the code
-RUN python init.py
-
 # run the main script
-CMD ["python", "main.py"]
+CMD ["python", "init.py"]

init.py

@@ -1,4 +1,5 @@
 import os
+import subprocess
 
 from huggingface_hub import login, snapshot_download
 
@@ -15,4 +16,11 @@ snapshot_download(
     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"])