app.py

import streamlit as st
from transformers import pipeline

# Charger le pipeline de génération de texte avec DistilBERT
generator = pipeline("text-generation", model="distilbert-base-uncased")

st.title("Générateur de texte avec DistilBERT")

# Zone de texte pour l'entrée de l'utilisateur
user_input = st.text_area("Entrez un texte pour générer la suite :")

if user_input:
    # Générer le texte
    generated_text = generator(user_input, max_length=50, num_return_sequences=1)
    st.write(generated_text[0]['generated_text'])