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
Files changed (1) hide show
  1. 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="text",
 
 
 
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."