Update app.py

app.py

@@ -3,20 +3,12 @@ from transformers import pipeline
 
 # Titre de l'application
 st.set_page_config(page_title="Chatbot AI", page_icon="🤖")
-st.title("Chatbot AI avec Hugging Face - Mistral AI ou BlenderBot")
-
+st.title("Chatbot AI avec Hugging Face - GPT2")
+    
 # Chargement du modèle
 @st.cache_data
 def load_model():
-    # Si vous utilisez un modèle Mistral AI, mettez ici le modèle approprié
-    # Exemple de modèle de Mistral ou BlenderBot
-    try:
-        # Modèle Mistral (hypothétique ici, ajustez selon votre modèle Hugging Face)
-        return pipeline("text-generation", model="mistral-ai/mistral-7B", device=0)  # Utilisez `device=0` si vous avez un GPU
-    except Exception as e:
-        st.error(f"Erreur de chargement du modèle: {e}")
-        # Modèle de remplacement Blenderbot si Mistral échoue
-        return pipeline("text-generation", model="facebook/blenderbot-400M-distill")
+    return pipeline("text-generation", model="gpt2")
 
 model = load_model()