Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import gradio as gr | |
pipe_ar = pipeline("question-answering",model='ZeyadAhmed/AraElectra-Arabic-SQuADv2-QA') | |
pipe_en = pipeline("question-answering",model='deepset/roberta-base-squad2') | |
def q_a(lang,text,question): | |
if lang == 'Arabic': | |
myinput = { | |
'question': question, | |
'context':text | |
} | |
return pipe_ar(myinput)['answer'] | |
elif lang == 'English': | |
myinput = { | |
'question': question, | |
'context':text | |
} | |
return pipe_en(myinput)['answer'] | |
demo = gr.Interface( | |
fn= q_a, | |
inputs=[gr.Radio(['Arabic', 'English'], label='Select Language',value= 'Arabic'), | |
gr.Textbox(label = 'enter text',lines=10), | |
gr.Textbox(label = 'enter question')], | |
outputs=gr.Textbox(label = 'answer') | |
) | |
demo.launch() |