Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
 from transformers import pipeline
 
-# Charger le modèle de génération de texte
+# Charger un modèle plus léger
 @st.cache_resource
 def load_model():
-    return pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
+    return pipeline("text-generation", model="distilgpt2")
 
 bot = load_model()
 
@@ -15,5 +15,5 @@ st.write("Posez vos questions sur le restaurant !")
 user_input = st.text_input("Votre question :")
 
 if user_input:
-    response = bot(user_input, max_length=200, do_sample=True)[0]["generated_text"]
+    response = bot(user_input, max_length=100, do_sample=True)[0]["generated_text"]
     st.write(response)