File size: 2,077 Bytes
9a2d345 be0043c 8e54fd5 de867e6 eda3840 de867e6 772bcf5 a20588f 5c2ddc8 a20588f 8e54fd5 52f171c b0adbfb 52f171c b0adbfb 1f7b435 05ff5c6 4959ff3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
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() |