| 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 message sous forme de liste [user, bot] | |
| history.append([message, answer]) | |
| # Retourner la liste de listes | |
| return history | |
| # Interface Gradio 3.43 | |
| iface = gr.ChatInterface(respond, title="Chat Demo") | |
| iface.launch() | |