Spaces:
Sleeping
Sleeping
File size: 1,683 Bytes
0947e33 56545ca 0947e33 07f8cab 2bebc3a 0947e33 |
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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import gradio as gr
from huggingface_hub import InferenceClient
model_name = "gpt2-large"
inference = InferenceClient(
model=model_name,
# token=os.environ['HF_API_KEY']
)
def respond(message, chat_history, max_tokens=32):
bot_message = inference.text_generation(
prompt=message,
max_new_tokens=max_tokens,
stop_sequences=['.'], # Stop generating tokens if a member of stop_sequences is generated
)
chat_history.append((message, f"{bot_message}."))
return "", chat_history
with gr.Blocks(
title='RugbyXpert',
# theme='sudeepshouche/minimalist', # https://www.gradio.app/guides/theming-guide
) as demo:
gr.Markdown(
"""
# RugbyXpert
"""
)
chatbot = gr.Chatbot(
height=310, # just to fit the notebook
)
msg = gr.Textbox(label="Pose-moi une question sur le rugby pendant la saison 2022-2023")
with gr.Row():
with gr.Column():
btn = gr.Button("Submit", variant="primary")
with gr.Column():
clear = gr.ClearButton(components=[msg, chatbot], value="Clear console")
gr.Examples([
"Retrouve-moi le nom du stade lors du match opposant Racing 92 à Castres le samedi 03 septembre 2022 ?",
"Tu peux me dire le 10 de Soyaux-Angoulême lors du match les opposant à Aurillac du vendredi 09 septembre 2022 ?",
"Dis-moi le poste de Kwagga Smith lors du match opposant Afrique du Sud à Italie du samedi 19 novembre 2022 ?",
], [msg])
btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot]) #Press enter to submit
demo.launch() |