my-rugbyxpert / app.py
nico-che's picture
Create app.py
69e1b87
raw
history blame
2.17 kB
import gradio as gr
from huggingface_hub import InferenceClient
model_name = "fake"
# inference = InferenceClient(
# model=model_name,
# # token=os.environ['HF_API_KEY']
# )
def respond(message, chat_history, max_tokens=32):
if "le nom du stade lors du match opposant Racing 92 à Castres le samedi 03 septembre 2022" in message:
bot_message = "Le nom du stade est Paris La Défense Arena."
elif "le 10 de Soyaux-Angoulême lors du match les opposant à Aurillac du vendredi 09 septembre 2022" in message:
bot_message = "Le 10 de Soyaux-Angoulême était Jacob Botica lors de ce match."
elif "le poste de Kwagga Smith lors du match opposant Afrique du Sud à Italie du samedi 19 novembre 2022" in message:
bot_message = "Le poste de Kwagga Smith était troisième ligne lors de ce match."
else:
bot_message = "Je n'ai pas compris. Veuillez reformuler votre question"
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()