Update app.py

app.py

@@ -1,17 +1,16 @@
 import streamlit as st
 from transformers import pipeline
 
-# Charger le modèle via le pipeline
-pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
+# Charger le pipeline de génération de texte avec DistilBERT
+generator = pipeline("text-generation", model="distilbert-base-uncased")
 
-# Titre de l'application Streamlit
-st.title("DeepSeek-R1 Text Generation")
+st.title("Générateur de texte avec DistilBERT")
 
-# Zone de saisie de texte pour l'utilisateur
-user_input = st.text_area("Entrez votre texte pour la génération", "")
+# Zone de texte pour l'entrée de l'utilisateur
+user_input = st.text_area("Entrez un texte pour générer la suite :")
 
-# Si un texte est entré, générer du texte
 if user_input:
-    output = pipe(user_input)
-    st.write("Texte généré :")
-    st.write(output[0]["generated_text"])
+    # Générer le texte
+    generated_text = generator(user_input, max_length=50, num_return_sequences=1)
+    st.write(generated_text[0]['generated_text'])
+