| import streamlit as st |
| from transformers import pipeline |
|
|
| |
| st.title("Générateur de Texte avec Hugging Face") |
|
|
| |
| model_name = st.selectbox("Choisissez un modèle :", ["gpt2", "distilgpt2"]) |
|
|
| |
| text_generator = pipeline("text-generation", model=model_name) |
|
|
| |
| prompt = st.text_area("Entrez votre prompt :", "Écrivez ici...") |
|
|
| |
| if st.button("Générer"): |
| if prompt: |
| with st.spinner("Génération en cours..."): |
| generated_text = text_generator(prompt, max_length=50, num_return_sequences=1)[0]['generated_text'] |
| st.success("Texte généré :") |
| st.write(generated_text) |
| else: |
| st.error("Veuillez entrer un prompt.") |
|
|
| |
| if __name__ == "__main__": |
| st.run() |
|
|
