import streamlit as st
from transformers import pipeline

# Initialisation du pipeline de génération de texte
generator = pipeline("text-generation", model="distilgpt2")

# Interface utilisateur Streamlit
st.title("Générateur de texte avec distilgpt2")
user_input = st.text_input("Entrez un texte pour générer la suite :")

if user_input:
    generated_text = generator(user_input, max_length=100, num_return_sequences=1)
    st.write(generated_text[0]['generated_text'])