MarcBrun commited on
Commit
a20588f
·
1 Parent(s): ef01d09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -11
app.py CHANGED
@@ -10,21 +10,16 @@ qa1 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad")
10
  qa2 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad-eu")
11
  qa3 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad-eu-en")
12
 
13
- def answer_question(question, context, pipe):
14
 
15
- if pipe == "qa1":
16
  answer = qa1(question=question, context=context)
17
- elif pipe == "qa2":
18
- answer = qa2(question=question, context=context)
19
- else:
20
- answer = qa3(question=question, context=context)
21
 
22
- return [answer["answer"], answer["score"]]
23
 
24
- iface1 = gr.Interface(fn=answer_question(pipe="qa1"), inputs=[gr.inputs.Textbox(lines=1, default=question, placeholder="Question Here...", label="Question"),gr.inputs.Textbox(lines=7, default=context, placeholder="Context Here...", label="Context")], outputs=[gr.outputs.Textbox(label="Answer"),gr.outputs.Textbox(label="Score")], capture_session=True)
25
 
26
- iface2 = gr.Interface(fn=answer_question(pipe="qa2"), inputs=[gr.inputs.Textbox(lines=1, default=question, placeholder="Question Here...", label="Question"),gr.inputs.Textbox(lines=7, default=context, placeholder="Context Here...", label="Context")], outputs=[gr.outputs.Textbox(label="Answer"),gr.outputs.Textbox(label="Score")], capture_session=True)
27
 
28
- iface3 = gr.Interface(fn=answer_question(pipe="qa3"), inputs=[gr.inputs.Textbox(lines=1, default=question, placeholder="Question Here...", label="Question"),gr.inputs.Textbox(lines=7, default=context, placeholder="Context Here...", label="Context")], outputs=[gr.outputs.Textbox(label="Answer"),gr.outputs.Textbox(label="Score")], capture_session=True, title="Question Answering in English, Spanish and Basque", description="Write or paste a text and a question about something that appears in the text and see if the model can answer correctly. It works best in English, but it has proven itself to also understand Spanish and Basque!\n\nI offer you a default question and answer for qick testing, but feel free to use your own examples.")
29
 
30
- Parallel(iface1,iface2, iface3).launch()
 
10
  qa2 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad-eu")
11
  qa3 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad-eu-en")
12
 
13
+ def answer_question1(question, context):
14
 
 
15
  answer = qa1(question=question, context=context)
 
 
 
 
16
 
17
+ return answer["answer"]
18
 
19
+ iface1 = gr.Interface(fn=answer_question1, inputs=[gr.inputs.Textbox(),gr.inputs.Textbox()], outputs=gr.outputs.Textbox(label="ixambert-squad"), capture_session=True)
20
 
21
+ iface2 = gr.Interface(fn=answer_question2, inputs=[gr.inputs.Textbox(),gr.inputs.Textbox()], outputs=gr.outputs.Textbox(label="ixambert-squad-eu"), capture_session=True)
22
 
23
+ iface3 = gr.Interface(fn=answer_question3, outputs=gr.outputs.Textbox(label="ixambert-squad-eu-en"), capture_session=True)
24
 
25
+ Parallel(iface1, iface2, iface3, title="Question Answering in English, Spanish and Basque", inputs=[gr.inputs.Textbox(lines=1, default=question, placeholder="Question Here...", label="Question"),gr.inputs.Textbox(lines=7, default=context, placeholder="Context Here...", label="Context")],description="Write or paste a text and a question about something that appears in the text and see if the model can answer correctly. It works best in English, but it has proven itself to also understand Spanish and Basque!\n\nI offer you a default question and answer for qick testing, but feel free to use your own examples.").launch()