File size: 1,182 Bytes
c37ca4f 1b8ae84 c37ca4f 1b8ae84 c37ca4f 1b8ae84 c37ca4f 1b8ae84 c37ca4f 1b8ae84 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import streamlit as st
import subprocess
def prepare_data():
st.text("Préparation des données en cours...")
subprocess.run(["python", "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"])
st.text("Entraînement du modèle terminé.")
def generate_samples():
st.text("Génération d'échantillons en cours...")
subprocess.run(["python", "sample.py", "--out_dir=out-shakespeare-char", "--device=cpu"])
st.text("Génération d'échantillons terminée.")
def main():
st.title("Application de Commandes")
if st.button("Préparer les données"):
prepare_data()
if st.button("Entraîner le modèle"):
train_model()
if st.button("Générer des échantillons"):
generate_samples()
if __name__ == "__main__":
main()
|