Spaces:
Runtime error
Runtime error
update qa
Browse files
app.py
CHANGED
@@ -176,13 +176,12 @@ pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
|
|
176 |
|
177 |
def sim(question, num_beams, text):
|
178 |
prefix = "[Frais irrépétibles ; article 700]"
|
179 |
-
q_type = 0
|
180 |
num_beams = None if num_beams == 0 else num_beams
|
181 |
-
if
|
182 |
-
question = "Qui sont les demandeurs initiaux ?"
|
183 |
-
elif
|
184 |
-
question = "Qui sont les demandeurs
|
185 |
-
q_type = 1
|
186 |
|
187 |
text = prefix + " " + question + " <sep> " + text
|
188 |
generated_text = pipe(
|
@@ -198,6 +197,7 @@ def sim(question, num_beams, text):
|
|
198 |
generated_tokens = generated_text[0]["generated_token_ids"]
|
199 |
tokens = generated_tokens.tolist()
|
200 |
sentence = tokenizer.decode(tokens).replace("<s>", "").replace("</s>", "").strip()
|
|
|
201 |
if q_type == 0:
|
202 |
sentence = sentence.split("<liste>")
|
203 |
return "Demandeurs initiaux :\n" + "\n".join(sentence)
|
@@ -215,7 +215,7 @@ def sim(question, num_beams, text):
|
|
215 |
iface = gr.Interface(
|
216 |
sim,
|
217 |
inputs=[
|
218 |
-
gr.Dropdown(["Demandeur initial", "Récapitulatif des demandes"], label="Information à générer", info="Succeptible de changer."),
|
219 |
gr.Slider(label="Nombre de Beams", minimum=0, maximum=10, value=5, step=1),
|
220 |
gr.Textbox(label= "Décision de justice", placeholder="Décision à traiter...", lines=25)
|
221 |
],
|
|
|
176 |
|
177 |
def sim(question, num_beams, text):
|
178 |
prefix = "[Frais irrépétibles ; article 700]"
|
179 |
+
q_type = int(question[0])
|
180 |
num_beams = None if num_beams == 0 else num_beams
|
181 |
+
if q_type == 0:
|
182 |
+
question = "Qui sont les demandeurs initiaux (ceux qui assignent) ?"
|
183 |
+
elif q_type == 1:
|
184 |
+
question = "Qui sont les demandeurs, le résultat de chaque demande et les quanta obtenus?"
|
|
|
185 |
|
186 |
text = prefix + " " + question + " <sep> " + text
|
187 |
generated_text = pipe(
|
|
|
197 |
generated_tokens = generated_text[0]["generated_token_ids"]
|
198 |
tokens = generated_tokens.tolist()
|
199 |
sentence = tokenizer.decode(tokens).replace("<s>", "").replace("</s>", "").strip()
|
200 |
+
|
201 |
if q_type == 0:
|
202 |
sentence = sentence.split("<liste>")
|
203 |
return "Demandeurs initiaux :\n" + "\n".join(sentence)
|
|
|
215 |
iface = gr.Interface(
|
216 |
sim,
|
217 |
inputs=[
|
218 |
+
gr.Dropdown(["0: Demandeur initial", "1: Récapitulatif des demandes"], label="Information à générer", info="Succeptible de changer."),
|
219 |
gr.Slider(label="Nombre de Beams", minimum=0, maximum=10, value=5, step=1),
|
220 |
gr.Textbox(label= "Décision de justice", placeholder="Décision à traiter...", lines=25)
|
221 |
],
|