Update app.py

app.py

@@ -15,5 +15,11 @@ st.write("Posez vos questions sur le restaurant !")
 user_input = st.text_input("Votre question :")
 
 if user_input:
-    response = bot(user_input, max_length=100, do_sample=True)[0]["generated_text"]
+    # Ajouter un prompt pour guider le modèle
+    prompt = f"Tu es un chatbot pour un restaurant. Réponds aux questions de manière amicale et informative. Voici la question : {user_input}"
+    
+    # Générer la réponse du bot
+    response = bot(prompt, max_length=100, do_sample=True)[0]["generated_text"]
+    
+    # Afficher la réponse
     st.write(response)