test_t5

app.py

@@ -3,12 +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 - Zephyr 7B Beta")
+st.title("Chatbot AI avec Hugging Face - Flan-T5 Large")
 
 # Chargement du modèle
-@st.cache(allow_output_mutation=True)
+@st.cache_data
 def load_model():
-    return pipeline("text-generation", model="zephyr-7b-beta")
+    return pipeline("text2text-generation", model="google/flan-t5-large")
 
 model = load_model()
 
@@ -18,7 +18,7 @@ user_input = st.text_input("Vous : ", "")
 
 if st.button("Envoyer"):
     if user_input:
-        with st.spinner("Zephyr réfléchit..."):
+        with st.spinner("Flan-T5 réfléchit..."):
             # Génération de la réponse
             response = model(user_input, max_length=100, num_return_sequences=1)
             bot_response = response[0]['generated_text']
@@ -45,6 +45,7 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
+
 # import streamlit as st
 
 # x = st.slider('Select a value')