File size: 1,925 Bytes
f718006 c37ca4f 1b8ae84 2c74034 1b8ae84 2c74034 1b8ae84 d55a5a5 ce1c45c d55a5a5 ce1c45c d55a5a5 ce1c45c d55a5a5 c37ca4f dc382c5 |
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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
# -*- coding: utf-8 -*-
import streamlit as st
import subprocess
def prepare_data():
st.text("Préparation des données en cours...")
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"])
st.text("Entraînement du modèle terminé.")
def run_command(command):
try:
result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, text=True)
return result.stdout, result.stderr
except Exception as e:
return None, str(e)
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.title("Sortie de la commande Python dans Streamlit")
# Commande Python que vous souhaitez exécuter
command = "python sample.py --out_dir=out-shakespeare-char"
# Exécutez la commande et obtenez la sortie
stdout, stderr = run_command(command)
# Affichez la sortie dans Streamlit
st.subheader("Sortie standard (stdout) :")
st.code(stdout)
st.subheader("Sortie d'erreur (stderr) :")
st.code(stderr)
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()
|