Update app.py

app.py

@@ -1,16 +1,14 @@
 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")
+# Initialisation du pipeline de génération de texte
+generator = pipeline("text-generation", model="distilgpt2")
 
-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 :")
+# 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:
-    # Générer le texte
-    generated_text = generator(user_input, max_length=50, num_return_sequences=1)
+    generated_text = generator(user_input, max_length=100, num_return_sequences=1)
     st.write(generated_text[0]['generated_text'])