| import gradio as gr | |
| from gradio.mix import Parallel | |
| from transformers import pipeline | |
| import torch | |
| question = "¿Cuál es la fórmula más famosa de Einstein?" | |
| context = "En 1905, cuando Albert Einstein era un joven físico desconocido, empleado en la Oficina de Patentes de Berna, publicó su teoría de la relatividad especial. En ella incorporó, en un marco teórico simple fundamentado en postulados físicos sencillos, conceptos y fenómenos estudiados antes por Henri Poincaré y Hendrik Lorentz. Como una consecuencia lógica de esta teoría, dedujo la ecuación de la física más conocida a nivel popular: la equivalencia masa-energía, E=mc². Ese año publicó otros trabajos que sentarían algunas de las bases de la física estadística y de la mecánica cuántica." | |
| qa1 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad") | |
| qa2 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad-eu") | |
| qa3 = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad-eu-en") | |
| def answer_question1(question, context): | |
| answer = qa1(question=question, context=context) | |
| return answer["answer"] | |
| def answer_question2(question, context): | |
| answer = qa2(question=question, context=context) | |
| return answer["answer"] | |
| def answer_question3(question, context): | |
| answer = qa3(question=question, context=context) | |
| return answer["answer"] | |
| iface1 = gr.Interface(fn=answer_question1, inputs=[gr.inputs.Textbox(),gr.inputs.Textbox()], outputs=gr.outputs.Textbox(label="ixambert-squad")) | |
| iface2 = gr.Interface(fn=answer_question2, inputs=[gr.inputs.Textbox(),gr.inputs.Textbox()], outputs=gr.outputs.Textbox(label="ixambert-squad-eu")) | |
| iface3 = gr.Interface(fn=answer_question3, inputs=[gr.inputs.Textbox(),gr.inputs.Textbox()], outputs=gr.outputs.Textbox(label="ixambert-squad-eu-en")) | |
| Parallel(iface1, iface2, iface3, 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")]).launch() |