| 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() | |