27
Browse files- app/app.py +3 -3
- 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(["
|
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(["
|
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 = "
|
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
|
|