Tbruand
commited on
Commit
·
669937f
1
Parent(s):
aac811b
feat(interface): ajout d'une option pour sélectionner le type de modèle (zero-shot ou few-shot)
Browse files- app/interface.py +4 -1
app/interface.py
CHANGED
@@ -4,7 +4,10 @@ from app.handler import predict
|
|
4 |
def launch_app():
|
5 |
iface = gr.Interface(
|
6 |
fn=predict,
|
7 |
-
inputs=
|
|
|
|
|
|
|
8 |
outputs="markdown",
|
9 |
title="🧪 ToxiCheck",
|
10 |
description="Entrez un texte pour détecter s'il est toxique. Résultat avec score de confiance pour chaque label."
|
|
|
4 |
def launch_app():
|
5 |
iface = gr.Interface(
|
6 |
fn=predict,
|
7 |
+
inputs=[
|
8 |
+
gr.Textbox(label="Texte à analyser"),
|
9 |
+
gr.Dropdown(choices=["zero-shot", "few-shot"], label="Type de modèle", value="zero-shot")
|
10 |
+
],
|
11 |
outputs="markdown",
|
12 |
title="🧪 ToxiCheck",
|
13 |
description="Entrez un texte pour détecter s'il est toxique. Résultat avec score de confiance pour chaque label."
|