LDJA commited on
Commit
7715886
1 Parent(s): 9a811a1
Files changed (2) hide show
  1. app/app.py +3 -3
  2. app/requirements.txt +1 -2
app/app.py CHANGED
@@ -4,12 +4,12 @@ import subprocess
4
 
5
  def prepare_data():
6
  st.text("Préparation des données en cours...")
7
- subprocess.run(["python", "../data/shakespeare_char/prepare.py"])
8
  st.text("Préparation des données terminée.")
9
 
10
  def train_model():
11
  st.text("Entraînement du modèle en cours...")
12
- 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"])
13
  st.text("Entraînement du modèle terminé.")
14
 
15
  def run_command(command):
@@ -25,7 +25,7 @@ def generate_samples():
25
  st.title("Sortie de la commande Python dans Streamlit")
26
 
27
  # Commande Python que vous souhaitez exécuter
28
- command = "python sample.py --out_dir=out-shakespeare-char"
29
 
30
  # Exécutez la commande et obtenez la sortie
31
  stdout, stderr = run_command(command)
 
4
 
5
  def prepare_data():
6
  st.text("Préparation des données en cours...")
7
+ subprocess.run(["python3", "../data/shakespeare_char/prepare.py"])
8
  st.text("Préparation des données terminée.")
9
 
10
  def train_model():
11
  st.text("Entraînement du modèle en cours...")
12
+ subprocess.run(["python3", "../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"])
13
  st.text("Entraînement du modèle terminé.")
14
 
15
  def run_command(command):
 
25
  st.title("Sortie de la commande Python dans Streamlit")
26
 
27
  # Commande Python que vous souhaitez exécuter
28
+ command = "python3 sample.py --out_dir=out-shakespeare-char"
29
 
30
  # Exécutez la commande et obtenez la sortie
31
  stdout, stderr = run_command(command)
app/requirements.txt CHANGED
@@ -3,5 +3,4 @@ numpy
3
  transformers
4
  wandb
5
  tqdm
6
- streamlit
7
- python3
 
3
  transformers
4
  wandb
5
  tqdm
6
+ streamlit