Update app.py

app.py

@@ -3,12 +3,20 @@ 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 - DialoGPT")
+st.title("Chatbot AI avec Hugging Face - Mistral AI ou BlenderBot")
 
 # Chargement du modèle
 @st.cache_data
 def load_model():
-    return pipeline("text-generation", model="microsoft/DialoGPT-medium")
+    # 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")
 
 model = load_model()
 
@@ -18,7 +26,7 @@ user_input = st.text_input("Vous : ", "")
 
 if st.button("Envoyer"):
     if user_input:
-        with st.spinner("DialoGPT réfléchit..."):
+        with st.spinner("Réflexion de l'IA..."):
             # Génération de la réponse
             response = model(user_input, max_length=100, num_return_sequences=1)
             bot_response = response[0]['generated_text']
@@ -46,8 +54,6 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
-
-
 # import streamlit as st
 
 # x = st.slider('Select a value')