25

Dockerfile

@@ -4,7 +4,7 @@ LABEL Version="1.0"
 
 RUN apt-get update -y
 
-RUN apt-get install -y python3-pip python3 build-essential
+RUN apt-get install -y python3-pip python3-dev build-essential python3
 
 COPY ./app /app
 

app/app.py

@@ -4,12 +4,12 @@ import subprocess
 
 def prepare_data():
     st.text("Préparation des données en cours...")
-    subprocess.run(["python", "prepare.py"])
+    subprocess.run(["python", "../data/shakespeare_char/prepare.py"])
     st.text("Préparation des données terminée.")
 
 def train_model():
     st.text("Entraînement du modèle en cours...")
-    subprocess.run(["python", "train.py", "config/train_shakespeare_char.py", "--device=cpu", "--compile=False", "--eval_iters=20", "--log_interval=1", "--block_size=64", "--batch_size=12", "--n_layer=4", "--n_head=4", "--n_embd=128", "--max_iters=2000", "--lr_decay_iters=2000", "--dropout=0.0"])
+    subprocess.run(["python", "../train.py", "config/train_shakespeare_char.py", "--device=cpu", "--compile=False", "--eval_iters=20", "--log_interval=1", "--block_size=64", "--batch_size=12", "--n_layer=4", "--n_head=4", "--n_embd=128", "--max_iters=2000", "--lr_decay_iters=2000", "--dropout=0.0"])
     st.text("Entraînement du modèle terminé.")
 
 def run_command(command):