Tbruand
feat(interface): ajout d'une option pour sélectionner le type de modèle (zero-shot ou few-shot)
669937f
| import gradio as gr | |
| from app.handler import predict | |
| def launch_app(): | |
| iface = gr.Interface( | |
| fn=predict, | |
| inputs=[ | |
| gr.Textbox(label="Texte à analyser"), | |
| gr.Dropdown(choices=["zero-shot", "few-shot"], label="Type de modèle", value="zero-shot") | |
| ], | |
| outputs="markdown", | |
| title="🧪 ToxiCheck", | |
| description="Entrez un texte pour détecter s'il est toxique. Résultat avec score de confiance pour chaque label." | |
| ) | |
| iface.launch() |