import gradio as gr
title = "GPT-J-6B (Boris)"
description = "Démo Gradio pour GPT-J 6B (Boris), un transformer entraîné à l'aide du Mesh Transformer JAX de Ben Wang. GPT-J' fait référence à la classe du modèle, tandis que '6B' représente le nombre de paramètres entraînables. Pour l'utiliser, il suffit d'ajouter votre texte, ou de cliquer sur l'un des exemples pour le charger. Pour en savoir plus, consultez les liens ci-dessous."
article = "<p style='text-align: center'><a href='https://github.com/kingoflolz/mesh-transformer-jax' target='_blank'>GPT-J-6B: A 6 Billion Parameter Autoregressive Language Model</a></p>"
examples = [
["La tour mesure 324 mètres (1 063 pieds) de haut,"],
["L'orbite de la Lune autour de la Terre a"],
["Le bassin lisse de Borealis dans l'hémisphère nord couvre 40 %"],
]
gr.Interface.load(
"huggingface/asi/gpt-fr-cased-small",
inputs=gr.Textbox(lines=5, label="Texte d'entrée"),
title=title,
description=description,
article=article,
examples=examples,
enable_queue=True,
).launch()