File size: 547 Bytes
08d9250 2b79c44 cd49c11 bb85704 2b79c44 cd49c11 2b79c44 bb85704 2b79c44 bb85704 c529a92 bb85704 c529a92 2b79c44 cd49c11 c529a92 2b79c44 cd49c11 7fdf645 c529a92 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | 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()
|