Update app.py

app.py

@@ -1,10 +1,9 @@
-import streamlit as st
 from transformers import pipeline
 
-# Charger le modèle CamemBERT pour le français
+# Charger un modèle spécifique au français
 @st.cache_resource
 def load_model():
-    return pipeline("text-generation", model="camembert-base")
+    return pipeline("text-generation", model="flaubert/flaubert_base_cased")
 
 # Créer l'interface Streamlit
 st.title("Chatbot Restaurant 🍽️")
@@ -16,10 +15,10 @@ user_input = st.text_input("Votre question :")
 # Si l'utilisateur a posé une question
 if user_input:
     bot = load_model()  # Charger le modèle
-    prompt = f"Tu es un chatbot pour un restaurant. Réponds de manière amicale et polie, et assure-toi que ta réponse est claire et bien structurée. Voici la question : {user_input}"
+    prompt = f"Tu es un chatbot pour un restaurant. Réponds de manière amicale, polie et précise. Voici la question : {user_input}"
 
     # Générer la réponse du modèle
-    response = bot(prompt, max_length=100, do_sample=True)[0]["generated_text"]
+    response = bot(prompt, max_length=50, do_sample=False)[0]["generated_text"]
 
     # Afficher la réponse
     st.write(response)