update qa

app.py

@@ -176,13 +176,12 @@ pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 def sim(question, num_beams, text):
     prefix = "[Frais irrépétibles ; article 700]"
-    q_type = 0
+    q_type = int(question[0])
     num_beams = None if num_beams == 0 else num_beams
-    if question == "Demandeur initial":
-        question = "Qui sont les demandeurs initiaux ?"
-    elif question == "Récapitulatif des demandes":
-        question = "Qui sont les demandeurs à l'origine de l'assignation, le résultat de leur demande et le montant obtenu ?"
-        q_type = 1
+    if q_type == 0:
+        question = "Qui sont les demandeurs initiaux (ceux qui assignent) ?"
+    elif q_type == 1:
+        question = "Qui sont les demandeurs, le résultat de chaque demande et les quanta obtenus?"
     
     text = prefix + " " + question + " <sep> " + text
     generated_text = pipe(
@@ -198,6 +197,7 @@ def sim(question, num_beams, text):
     generated_tokens = generated_text[0]["generated_token_ids"]
     tokens = generated_tokens.tolist()
     sentence = tokenizer.decode(tokens).replace("<s>", "").replace("</s>", "").strip()
+    
     if q_type == 0:
         sentence = sentence.split("<liste>")
         return "Demandeurs initiaux :\n" + "\n".join(sentence)
@@ -215,7 +215,7 @@ def sim(question, num_beams, text):
 iface = gr.Interface(
     sim, 
     inputs=[
-        gr.Dropdown(["Demandeur initial", "Récapitulatif des demandes"], label="Information à générer", info="Succeptible de changer."), 
+        gr.Dropdown(["0: Demandeur initial", "1: Récapitulatif des demandes"], label="Information à générer", info="Succeptible de changer."), 
         gr.Slider(label="Nombre de Beams", minimum=0, maximum=10, value=5, step=1), 
         gr.Textbox(label= "Décision de justice", placeholder="Décision à traiter...", lines=25)
         ],