app.py

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"), capture_session=True)

iface2 = gr.Interface(fn=answer_question2, inputs=[gr.inputs.Textbox(),gr.inputs.Textbox()], outputs=gr.outputs.Textbox(label="ixambert-squad-eu"), capture_session=True)

iface3 = gr.Interface(fn=answer_question3, outputs=gr.outputs.Textbox(label="ixambert-squad-eu-en"), capture_session=True)

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()