| import gradio as gr | |
| from transformers import pipeline | |
| # GPT-2 public | |
| generator = pipeline("text-generation", model="gpt2") | |
| def respond(message, history): | |
| history = history or [] | |
| # Générer la réponse | |
| answer = generator(message, max_length=50, do_sample=True)[0]["generated_text"] | |
| # Ajouter le tuple (user, bot) à l'historique | |
| history.append((message, answer)) | |
| # Retourner la liste de tuples | |
| return history | |
| # Gradio 3.43 | |
| iface = gr.ChatInterface(respond, title="Chat Demo") | |
| iface.launch() | |