import streamlit as st
from transformers import pipeline

# Charger un modèle plus léger
@st.cache_resource
def load_model():
    return pipeline("text-generation", model="distilgpt2")

bot = load_model()

# Interface du chatbot
st.title("Chatbot Restaurant 🍽️")
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"]
    st.write(response)